Spaces:
Sleeping
Sleeping
rynmurdock
commited on
Commit
·
43c409b
1
Parent(s):
cf29a1f
Update app.py
Browse files
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=
|
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')
|