From 669562b71201ab81bc44ef8c4c53bf1565dbc605 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= <kimjh9903@naver.com>
Date: Fri, 27 Sep 2024 15:46:02 +0900
Subject: [PATCH] =?UTF-8?q?Fix:=20classification=20result=20=EC=88=98?=
 =?UTF-8?q?=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ai/app/api/yolo/classfication.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ai/app/api/yolo/classfication.py b/ai/app/api/yolo/classfication.py
index 59ca319..0accc43 100644
--- a/ai/app/api/yolo/classfication.py
+++ b/ai/app/api/yolo/classfication.py
@@ -118,7 +118,7 @@ async def classification_train(request: TrainRequest):
         recall= 0,
         mAP50= 0,
         mAP5095= 0,
-        accuracy=result["accuracy_top1"],
+        accuracy=result["metrics/accuracy_top1"],
         fitness= result["fitness"]
     )