rynmurdock commited on
Commit
43c409b
1 Parent(s): cf29a1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -206,7 +206,7 @@ def get_user_emb(embs, ys):
206
 
207
  print('Gathering coefficients')
208
  #lin_class = Ridge(fit_intercept=False).fit(feature_embs, chosen_y)
209
- lin_class = SVC(max_iter=500, kernel='linear', C=.1, class_weight='balanced').fit(feature_embs, chosen_y)
210
  coef_ = torch.tensor(lin_class.coef_, dtype=torch.double).detach().to('cpu')
211
  coef_ = coef_ / coef_.abs().max() * 3
212
  print('Gathered')
 
206
 
207
  print('Gathering coefficients')
208
  #lin_class = Ridge(fit_intercept=False).fit(feature_embs, chosen_y)
209
+ lin_class = SVC(max_iter=20, kernel='linear', C=.1, class_weight='balanced').fit(feature_embs, chosen_y)
210
  coef_ = torch.tensor(lin_class.coef_, dtype=torch.double).detach().to('cpu')
211
  coef_ = coef_ / coef_.abs().max() * 3
212
  print('Gathered')