首页 > 解决方案 > 如何修正图表

问题描述

使用 Python Scikit-learn 实现的 Logistic 回归算法,根据花瓣长度和宽度对 Iris 数据集中的三种花卉(Setosa、Versicolor、Virgin)进行分类。

我可以知道如何根据附件更正我的图表吗

from sklearn import datasets

#load data
iris=datasets.load_iris()
X=iris.data[:,[2,3]]
y=iris.target
from sklearn.model_selection import train_test_split
X_train, X_test,y_train, y_test=train_test_split(X,y,test_size=0.3, random_state=0,stratify=y)
#feature scaling
from sklearn.preprocessing import StandardScaler
sc=StandardScaler()
sc.fit(X_train)
X_train_std=sc.transform(X_train)
X_test_std=sc.transform(X_test)
#Logistic regression
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score
C1=[]
A=[]
C1=[0.01,0.1,1,10,100,1000]
for i in range(len(C1)):
    lr=LogisticRegression(C=C1[i], random_state=0)
    lr.fit(X_train_std,y_train)
    y_pred=lr.predict(X_test_std)
    A.append(accuracy_score(y_test,y_pred))

#draw figure
import matplotlib.pyplot as plt
plt.plot(C1,A)
plt.title('Logistic Regression')
plt.xlabel('C')
plt.ylabel('Accuracy')
plt.show()

看图片

在此处输入图像描述

如何修改编码以获得与所附图像文件相同的图片

标签: pythonmachine-learninglogistic-regression

解决方案


推荐阅读