机器学习中,跑实验常用到的一些API【自用,更新ing】

sklearn库中

1.混淆矩阵:

使用到的类:【confusion_matrix】,【ConfusionMatrixDisplay】
计算混淆矩阵:
from sklearn.metrics import confusion_matrix
画矩阵混淆图:
from sklearn.metrics import ConfusionMatrixDisplay
【参考链接:】

  1. https://scikit-learn.org/stable/auto_examples/model_selection/plot_confusion_matrix.html#sphx-glr-auto-examples-model-selection-plot-confusion-matrix-py
  2. 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

你可能感兴趣的:(机器学习,sklearn,机器学习,python)