worlabel/ai/app/schemas/train_response.py

14 lines
278 B
Python

from pydantic import BaseModel
class TrainResponse(BaseModel):
modelKey:str
precision: float
recall: float
mAP50: float
mAP5095: float
fitness: float
class ClassificationTrainResponse(BaseModel):
modelKey: str
accuracy: float
fitness: float