fix loop

aimclub · May 17, 2024 · a0ff806 · a0ff806
1 parent 8d092fc
commit a0ff806
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fedot_ind/tools/example_utils.py b/fedot_ind/tools/example_utils.py
@@ -115,7 +115,9 @@ def industrial_common_modelling_loop(
         industrial.fit(train_data)
 
     labels = industrial.predict(test_data)
-    probs = industrial.predict_proba(test_data)
+    if api_config['problem'] != 'ts_forecasting':
+        probs = industrial.predict_proba(test_data)
+
     metrics = industrial.get_metrics(target=test_data[1],
                                      rounding_order=3,
                                      metric_names=metric_names)