Mastouri commited on
Commit
599e887
·
1 Parent(s): ece7718

Resolve conflicts

Browse files
Files changed (1) hide show
  1. logistic_reg.py +1 -1
logistic_reg.py CHANGED
@@ -64,7 +64,7 @@ print(f"F1 Score: {classification_report(y_val_encoded, y_val_pred, target_names
64
  print(f"Hamming Loss: {hamming_loss(y_val_encoded, y_val_pred):.4f}")
65
 
66
  # Step 7: Save the Model and Preprocessing Artifacts
67
- dump(best_model, "optimized_logistic_model.joblib") # Save the optimized Logistic Regression model
68
  dump(tfidf_vectorizer, "tfidf_vectorizer.joblib") # Save the TF-IDF vectorizer
69
  dump(mlb, "label_binarizer.joblib") # Save the MultiLabelBinarizer
70
 
 
64
  print(f"Hamming Loss: {hamming_loss(y_val_encoded, y_val_pred):.4f}")
65
 
66
  # Step 7: Save the Model and Preprocessing Artifacts
67
+ dump(best_model, "logistic_model.joblib") # Save the optimized Logistic Regression model
68
  dump(tfidf_vectorizer, "tfidf_vectorizer.joblib") # Save the TF-IDF vectorizer
69
  dump(mlb, "label_binarizer.joblib") # Save the MultiLabelBinarizer
70