Fix loss function

nikiduki · nikiduki · commit aa45b90fe482 · 2025-05-14T17:38:18.000+03:00
diff --git a/autointent/modules/scoring/_catboost/catboost_scorer.py b/autointent/modules/scoring/_catboost/catboost_scorer.py
@@ -216,7 +216,7 @@ def fit(
             y = np.asarray(cast("Sequence[int]", labels), dtype=np.int64)
 
         default_loss = (
-            "MultiLabel"
+            "MultiLogloss"
             if self._multilabel
             else ("MultiClass" if self._n_classes > BINARY_CLASS_THRESHOLD else "Logloss")
         )
diff --git a/tests/assets/configs/multilabel.yaml b/tests/assets/configs/multilabel.yaml
@@ -30,7 +30,7 @@
       learning_rate: [0.05, 0.1]
       depth: [1, 10]
       l2_leaf_reg: [1, 5]
-      loss_function: ["MultiLabel"]
+      loss_function: ["MultiLogloss"]
       random_seed: [42]
       classification_model_config:
         - null

Original file line number	Diff line number	Diff line change
`@@ -216,7 +216,7 @@ def fit(`
`216`	`216`	`y = np.asarray(cast("Sequence[int]", labels), dtype=np.int64)`
`217`	`217`
`218`	`218`	`default_loss = (`
`219`		`- "MultiLabel"`
	`219`	`+ "MultiLogloss"`
`220`	`220`	`if self._multilabel`
`221`	`221`	`else ("MultiClass" if self._n_classes > BINARY_CLASS_THRESHOLD else "Logloss")`
`222`	`222`	`)`