文件名称：利用鸢尾花数据集画出P-R曲线 python
文件大小：3KB
文件格式：PY
更新时间：2023-04-08 04:27:10
机器学习 pr曲线 深度学习 python
#coding=utf-8 """ #演示目的：利用鸢尾花数据集画出P-R曲线 """ print(__doc__) import matplotlib.pyplot as plt import numpy as np from sklearn import svm, datasets from sklearn.metrics import precision_recall_curve from sklearn.metrics import average_precision_score from sklearn.preprocessing import label_binarize from sklearn.multiclass import OneVsRestClassifier #from sklearn.cross_validation import train_test_split #适用于anaconda 3.6及以前版本 from sklearn.model_selection import train_test_split#适用于anaconda 3.7 #以iris数据为例，画出P-R曲线 iris = datasets.load_iris() X = iris.data y = iris.target # 标签二值化,将三个类转为001, 010, 100的格式.因为这是个多类分类问题，后面将要采用 #OneVsRestClassifier策略转为二类分类问题 y = label_binarize(y, classes=[0, 1, 2]) n_classes = y.shape[1] print (y)