Spaces:

Devikalalitha
/

news_backend_services_monitor

Runtime error

Devikalalitha commited on Jul 7, 2024

Commit

205bd20

verified ·

1 Parent(s): 207d171

Update classification_model_monitor.py

Files changed (1) hide show

classification_model_monitor.py CHANGED Viewed

@@ -27,9 +27,9 @@ def filter_prediction_data(data: pd.DataFrame):
         # filtered_prediction_data = data.loc[data['y_true_proba'] == 1].copy()
-        filtered_prediction_data = data.loc[(data['y_true_proba'] == 1) & (data['used_for_training'].str.contains("_train")==False) &
-                                                                          (data['used_for_training'].str.contains("_excluded")==False) &
-                                                                          (data['used_for_training'].str.contains("_validation")==False),
                                             ['y_true', 'y_pred']].copy()
         logging.info("Exiting filter_prediction_data()")

         # filtered_prediction_data = data.loc[data['y_true_proba'] == 1].copy()
+        filtered_prediction_data = data.loc[(data['y_true_proba'] == 1) & (data['used_for_training'].astype("str").str.contains("_train")==False) &
+                                                                          (data['used_for_training'].astype("str").str.contains("_excluded")==False) &
+                                                                          (data['used_for_training'].astype("str").str.contains("_validation")==False),
                                             ['y_true', 'y_pred']].copy()
         logging.info("Exiting filter_prediction_data()")