pvyas96 commited on
Commit
4b43d6e
·
verified ·
1 Parent(s): 64ed970

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ def precision_at_k(preds_topk, Y_true, k):
103
  hits += len(set(pred_labels) & true_labels)
104
  return hits / (total * k)
105
 
106
- def evaluate_rforest(ensemble, X_test, Y_test, dim=400):
107
  svd = TruncatedSVD(n_components=dim, random_state=42)
108
  X_test_reduced = svd.fit_transform(X_test)
109
 
 
103
  hits += len(set(pred_labels) & true_labels)
104
  return hits / (total * k)
105
 
106
+ def evaluate_rforest(ensemble, X_test, Y_test, dim=300):
107
  svd = TruncatedSVD(n_components=dim, random_state=42)
108
  X_test_reduced = svd.fit_transform(X_test)
109