def SVMTest(): clf_tfidf = joblib.load(modelFile) y_predicted_tfidf = clf_tfidf.predict(X_test_tfidf) accuracy_tfidf, precision_tfidf, recall_tfidf, f1_tfidf = get_metrics(y_test, y_predicted_tfidf) print("accuracy = %.6f, precision = %.6f, recall = %.6f, f1 = %.6f" % ( accuracy_tfidf, precision_tfidf, recall_tfidf, f1_tfidf)) # 评估 print("Precision, Recall, F1-Score and support") print(metrics.classification_report(y_test, y_predicted_tfidf, target_names=categories)) # 混淆矩阵 print("Confusion Matrix...") cm = metrics.confusion_matrix(y_test, y_predicted_tfidf) print(cm)
时间: 2024-04-19 19:27:11 浏览: 104
TCL.CLF.tar.gz_TCL.CLF_sourceinsight TCL语言包
这段代码是在SVMTest函数中进行模型评估的部分。首先,你打印了准确率(accuracy_tfidf)、精确率(precision_tfidf)、召回率(recall_tfidf)和F1值(f1_tfidf)。然后,你使用classification_report函数打印了分类报告,其中包括每个类别的精确率、召回率和F1值。接下来,你使用confusion_matrix函数计算了混淆矩阵(cm),并打印了该矩阵。混淆矩阵可以用于了解模型在每个类别上的分类情况。
阅读全文