classifier of chains

最新推荐文章于 2022-09-23 16:14:24 发布

weixin_30533797

最新推荐文章于 2022-09-23 16:14:24 发布

阅读量676

点赞数

文章标签：人工智能数据结构与算法 python

原文链接：http://www.cnblogs.com/muamu/p/11550717.html

版权

from skmultilearn.problem_transform import ClassifierChain
from sklearn.naive_bayes import GaussianNB
#scikit-learn version>=0.21 
from sklearn.metrics import multilabel_confusion_matrix
from scipy.sparse import csr_matrix
def svm():

　　clf = ClassifierChain(GaussianNB())
　　S = csr_matrix(train_labels)
　　#S为sprase matrix
　　clf.fit(train_features, S)

　　prediction = clf.predict(csr_matrix(test_features))

　　cnf_matrix = multilabel_confusion_matrix(y_true, y_prediction)
# print(cnf_matrix)

　　TN = cnf_matrix[:, 0, 0]
　　TP = cnf_matrix[:, 1, 1]
　　FN = cnf_matrix[:, 1, 0]
　　FP = cnf_matrix[:, 0, 1]

　　ACC = accuracy_score(y_true, y_prediction)
　　TPR =  metrics.recall_score(y_true, y_prediction, average='macro')
　　TNR =  np.mean(TN / (TN + FP))
　　F1_score =  metrics.f1_score(y_true, y_prediction, average='macro')

转载于:https://www.cnblogs.com/muamu/p/11550717.html