rynmurdock commited on
Commit
d23d2cd
1 Parent(s): e819e80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -191,7 +191,7 @@ def get_user_emb(embs, ys):
191
  print('ys are longer than embs; popping latest rating')
192
  ys.pop(-1)
193
 
194
- feature_embs = np.array(torch.stack([embs[i].squeeze().to('cpu') for i in indices]]).to('cpu'))
195
  #scaler = preprocessing.StandardScaler().fit(feature_embs)
196
  #feature_embs = scaler.transform(feature_embs)
197
  chosen_y = np.array([ys[i] for i in indices])
 
191
  print('ys are longer than embs; popping latest rating')
192
  ys.pop(-1)
193
 
194
+ feature_embs = np.array(torch.stack([embs[i].squeeze().to('cpu') for i in indices]).to('cpu'))
195
  #scaler = preprocessing.StandardScaler().fit(feature_embs)
196
  #feature_embs = scaler.transform(feature_embs)
197
  chosen_y = np.array([ys[i] for i in indices])