sklearn库中
1.混淆矩阵:
使用到的类:【confusion_matrix】,【ConfusionMatrixDisplay】
计算混淆矩阵:
from sklearn.metrics import confusion_matrix
画矩阵混淆图:
from sklearn.metrics import ConfusionMatrixDisplay
【参考链接:】
- https://scikit-learn.org/stable/auto_examples/model_selection/plot_confusion_matrix.html#sphx-glr-auto-examples-model-selection-plot-confusion-matrix-py
- https://scikit-learn.org/stable/modules/model_evaluation.html#confusion-matrix
2. 分类报告(每种类的准确率、召回率等信息)
使用到的类:【】
https://scikit-learn.org/stable/modules/model_evaluation.html#classification-report
3. 精确率、准确率、召回率、F1值
使用到的类【】
https://scikit-learn.org/stable/modules/model_evaluation.html#precision-recall-and-f-measures
4.分割训练集和测试集
from sklearn.model_selection import train_test_split