fix weight calculation if a model does not make predictions for all classes lists in its trust-weights file

Author: sfluegel05

chebifier/ensemble/weighted_majority_ensemble.py

@@ -45,13 +45,14 @@ def calculate_classwise_weights(self, predicted_classes):
             if model.classwise_weights is None:
                 continue
             for cls, weights in model.classwise_weights.items():
-                if (2 * weights["TP"] + weights["FP"] + weights["FN"]) > 0:
-                    f1 = (
-                        2
-                        * weights["TP"]
-                        / (2 * weights["TP"] + weights["FP"] + weights["FN"])
-                    )
-                    weights_by_cls[predicted_classes[cls], j] *= 1 + f1
+                if cls in predicted_classes:
+                    if (2 * weights["TP"] + weights["FP"] + weights["FN"]) > 0:
+                        f1 = (
+                            2
+                            * weights["TP"]
+                            / (2 * weights["TP"] + weights["FP"] + weights["FN"])
+                        )
+                        weights_by_cls[predicted_classes[cls], j] *= 1 + f1
 
         print("Calculated model weightings. The average weights are:")
         for i, model in enumerate(self.models):