fix: set optional field of current multiclass model quality (#77)

* fix: set optional field for current model quality of multiclass metrics * fix: (sdk) set optional field for current model quality of multiclass metrics
radicalbit · Jul 2, 2024 · 13b0135 · 13b0135
1 parent 70b2b42
commit 13b0135
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/app/models/metrics/model_quality_dto.py b/app/models/metrics/model_quality_dto.py
@@ -43,16 +43,16 @@ class Distribution(BaseModel):
 
 
 class GroupedMetricsBase(BaseModel):
-    f1: List[Distribution]
-    accuracy: List[Distribution]
+    f1: Optional[List[Distribution]] = None
+    accuracy: Optional[List[Distribution]] = None
     precision: List[Distribution]
     recall: List[Distribution]
     f_measure: List[Distribution]
-    weighted_precision: List[Distribution]
-    weighted_recall: List[Distribution]
-    weighted_f_measure: List[Distribution]
-    weighted_true_positive_rate: List[Distribution]
-    weighted_false_positive_rate: List[Distribution]
+    weighted_precision: Optional[List[Distribution]] = None
+    weighted_recall: Optional[List[Distribution]] = None
+    weighted_f_measure: Optional[List[Distribution]] = None
+    weighted_true_positive_rate: Optional[List[Distribution]] = None
+    weighted_false_positive_rate: Optional[List[Distribution]] = None
     true_positive_rate: List[Distribution]
     false_positive_rate: List[Distribution]
     area_under_roc: Optional[List[Distribution]] = None