Merge remote-tracking branch 'origin/tutorial_and_fixes' into tutoria…

…l_and_fixes
aimclub · Nov 28, 2024 · 5855e4a · 5855e4a
2 parents e685eab + d3c13b4
commit 5855e4a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/examples/automl_example/api_example/time_series/ts_classification/tmp.py b/examples/automl_example/api_example/time_series/ts_classification/tmp.py
@@ -21,8 +21,8 @@ def multiclass_classification():
     pdc.fit(X, y)
     print('score:', pdc.score(X, y))
 
-    y_pred = pdc.predict(X)
-    proba_pred = pdc.predict_proba(X)
+    pdc.predict(X)
+    pdc.predict_proba(X)
 
     assert pdc.score(X, y) == 1.0
 

diff --git a/fedot_ind/api/main.py b/fedot_ind/api/main.py
@@ -324,7 +324,7 @@ def get_metrics(self,
                     predicted_probs=probs,
                     rounding_order=rounding_order,
                     metric_names=metric_names) for strategy,
-                                                   probs in self.predicted_probs.items()}
+                probs in self.predicted_probs.items()}
 
         else:
             metric_dict = self._metric_evaluation_loop(

diff --git a/fedot_ind/core/repository/constanst_repository.py b/fedot_ind/core/repository/constanst_repository.py
@@ -409,6 +409,7 @@ class FedotOperationConstant(Enum):
         "catboostreg": FedotCatBoostRegressionImplementation
     }
 
+
 class ModelCompressionConstant(Enum):
     ENERGY_THR = [0.9, 0.95, 0.99, 0.999]
     DECOMPOSE_MODE = 'channel'