Update classification_model_monitor.py

classification_model_monitor.py

@@ -160,7 +160,13 @@ def get_misclassified_classes(data):
         misclassified_examples = data.loc[data['match'] == 0, ['text', 'y_true', 'y_pred', 'y_pred_proba', 'url']].copy()
         misclassified_examples.sort_values(by=['y_pred', 'y_pred_proba'], ascending=[True, False], inplace=True)
         
-        misclassifications = data.loc[data['match'] == 0, 'y_pred'].value_counts()[y_pred_counts.index]
+        misclassifications = data.loc[data['match'] == 0, 'y_pred'].value_counts()
+
+        missing_classes = [i for i in y_pred_counts.index if i not in misclassifications.index]
+        for i in missing_classes:
+            misclassifications[i] = 0
+        
+        misclassifications = misclassifications[y_pred_counts.index].copy()
         misclassifications /= y_pred_counts
         misclassifications.sort_values(ascending=False, inplace=True)
         logging.info("Exiting get_misclassified_classes()")