bioNLP-UNAM / useless

Go to a project
Toggle navigation Toggle navigation pinning
Carlos-Francisco Méndez-Cruz
5d820c1f 1 parent 98b2335f

Final version binClass for papers

Inline Side-by-side
Showing 1 changed file with 5 additions and 3 deletions
......@@ -102,9 +102,11 @@ if args.traind and not args.input:
if args.split:
X = vectorizer.transform(X_test)
y_pred = clf.predict(X)
print(precision_score(y_test, y_pred))
print(recall_score(y_test, y_pred))
print(f1_score(y_test, y_pred))
print("Test results: ")
print("Precision: {}".format(precision_score(y_test, y_pred)))
print("Recall: {}".format(recall_score(y_test, y_pred)))
print("F-score: {}".format(f1_score(y_test, y_pred)))
print("Classification report:")
print(classification_report(y_test, y_pred))
else:
from pdb import set_trace as st
......