lyangas commited on
Commit
88a4631
·
verified ·
1 Parent(s): 4585ef5

Update app.py

Browse files

print(clf_name)

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def classify_code(text, top_n):
69
  embed = [embedder(text)]
70
  preds = {}
71
  for clf_name in classifiers_codes.keys():
 
72
  model = classifiers_codes[clf_name]
73
  probs = model.predict_proba(embed)
74
  best_n = np.flip(np.argsort(probs, axis=1,)[0,-top_n:])
 
69
  embed = [embedder(text)]
70
  preds = {}
71
  for clf_name in classifiers_codes.keys():
72
+ print(clf_name)
73
  model = classifiers_codes[clf_name]
74
  probs = model.predict_proba(embed)
75
  best_n = np.flip(np.argsort(probs, axis=1,)[0,-top_n:])