2024-09-26 17:20:37 +09:00
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
|
|
|
|
class TrainResponse(BaseModel):
|
|
|
|
|
modelKey:str
|
|
|
|
|
precision: float
|
|
|
|
|
recall: float
|
|
|
|
|
mAP50: float
|
|
|
|
|
mAP5095: float
|
2024-09-27 14:08:43 +09:00
|
|
|
fitness: float
|
|
|
|
|
|
|
|
|
|
class ClassificationTrainResponse(BaseModel):
|
|
|
|
|
modelKey: str
|
|
|
|
|
accuracy: float
|
2024-09-26 17:20:37 +09:00
|
|
|
fitness: float
|
