Update classification_model_monitor.py

classification_model_monitor.py

@@ -22,6 +22,7 @@ def filter_prediction_data(data: pd.DataFrame):
             raise Exception("Input Prediction Data frame in None")   
 
         filtered_prediction_data = data.loc[(data['y_true'].isin(['WEATHER', 'EDUCATION', 'ASTROLOGY', 'OTHERS']) == False) & 
+                                            (data['y_pred'].isin(['WEATHER', 'EDUCATION', 'ASTROLOGY', 'OTHERS']) == False) &
                                             (data['y_true_proba'] > CLASSIFIER_THRESHOLD)].copy()
             
         logging.info("Exiting filter_prediction_data()")