Agnist commited on
Commit
88b7669
·
verified ·
1 Parent(s): 0dd1ac9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ except ValueError as e:
72
  X_train_resampled, y_train_resampled = X_train_tfidf, y_train
73
 
74
  # Logistic Regression Model
75
- model = LogisticRegression(C=10, max_iter=1000, n_jobs=-1, solver='lbfgs', multi_class='multinomial')
76
  model.fit(X_train_resampled, y_train_resampled)
77
 
78
  # Evaluate Model
 
72
  X_train_resampled, y_train_resampled = X_train_tfidf, y_train
73
 
74
  # Logistic Regression Model
75
+ model = LogisticRegression(C=1, max_iter=5000, n_jobs=-1, solver='lbfgs', multi_class='multinomial', class_weight='balanced')
76
  model.fit(X_train_resampled, y_train_resampled)
77
 
78
  # Evaluate Model