Update app.py
Browse filesprint(clf_name)
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:])
|