Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- classifier = OneVsRestClassifier(SGDClassifier(loss='log', alpha=0.00001, penalty='l1', class_weight="balanced"), n_jobs=-1)
- classifier.fit(x_train_multilabel, y_train)
- predictions = classifier.predict (x_test_multilabel)
- precision = precision_score(y_test, predictions, average='micro')
- recall = recall_score(y_test, predictions, average='micro')
- f1 = f1_score(y_test, predictions, average='micro')
- print("Micro-average quality numbers")
- print("Precision: {:.4f}, Recall: {:.4f}, F1-measure: {:.4f}".format(precision, recall, f1))
- precision = precision_score(y_test, predictions, average='macro')
- recall = recall_score(y_test, predictions, average='macro')
- f1 = f1_score(y_test, predictions, average='macro')
- print("Macro-average quality numbers")
- print("Precision: {:.4f}, Recall: {:.4f}, F1-measure: {:.4f}".format(precision, recall, f1))
- print (metrics.classification_report(y_test, predictions))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement