Update dataset_utils.py

dataset_utils.py

@@ -142,14 +142,21 @@ def save_label_encoders(label_encoders):
 def load_label_encoders():
     """
     Loads a dictionary of label encoders from a pickle file.
+    Handles scikit-learn version mismatches gracefully.
 
     Returns:
         dict: Loaded dictionary of LabelEncoder objects.
     """
-    with open(LABEL_ENCODERS_PATH, "rb") as f:
-        return pickle.load(f)
-    print(f"Label encoders loaded from {LABEL_ENCODERS_PATH}")
-
+    try:
+        with open(LABEL_ENCODERS_PATH, "rb") as f:
+            label_encoders = pickle.load(f)
+        print(f"Label encoders loaded from {LABEL_ENCODERS_PATH}")
+        return label_encoders
+    except Exception as e:
+        print(f"Warning: Could not load label encoders from {LABEL_ENCODERS_PATH}: {str(e)}")
+        print("Creating new label encoders...")
+        # Create empty label encoders as fallback
+        return {col: LabelEncoder() for col in LABEL_COLUMNS}
 
 def get_num_labels(label_encoders):
     """