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:])
|