correct target score

aimclub · Nov 20, 2023 · 7076b6b · 7076b6b
1 parent 31dcaec
commit 7076b6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/Machine_learning/SVC/_2D/Problems/SVC_2D_Transformators_State.py b/examples/Machine_learning/SVC/_2D/Problems/SVC_2D_Transformators_State.py
@@ -54,5 +54,5 @@ def calculate(self, point: Point, function_value: FunctionValue) -> FunctionValu
         """
         cs, gammas = point.float_variables[0], point.float_variables[1]
         clf = SVC(C=10 ** cs, gamma=10 ** gammas)
-        function_value.value = -cross_val_score(clf, self.x, self.y, scoring='accuracy').mean()
+        function_value.value = -cross_val_score(clf, self.x, self.y, scoring='f1_macro').mean()
         return function_value