osslab-pku
/
gfi-bot
mirror of https://github.com/osslab-pku/gfi-bot.git
7
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix(model): add batch_precision,recall,f1

This commit is contained in:
mcxwx 2022-06-15 18:31:08 +08:00 committed by GitHub
parent 2f917d69e9
commit 095a28ed29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gfibot/collections/model.py
View File

@ -61,6 +61,9 @@ class TrainingSummary(Document):
f1: float = FloatField(null=True)
batch_accuracy: List[str] = ListField(StringField(), default=[])
batch_auc: List[str] = ListField(StringField(), default=[])
batch_precision: List[str] = ListField(StringField(), default=[])
batch_recall: List[str] = ListField(StringField(), default=[])
batch_f1: List[str] = ListField(StringField(), default=[])
last_updated: datetime = DateTimeField(required=True)
meta = {
"indexes": [