使用交叉验证对鸢尾花分类模型进行调参课程代码.pdfVIP

#数据源地址：，压缩包后解压

即可（数据源与上节课相同）

#-*-coding:utf-8-*-

任务：鸢尾花识别

importpandasaspd

fromsklearn.model_selectionimporttrain_test_split,GridSearchCV

fromsklearn.neighborsimportKNeighborsClassifier

fromsklearn.linear_modelimportLogisticRegression

fromsklearn.svmimportSVC

DATA_FILE=./data_ai/Iris.csv

SPECIES_LABEL_DICT={

Iris-setosa:0,#山鸢尾

Iris-versicolor:1,#变色鸢尾

Iris-virginica:2#维吉尼亚鸢尾

}

#使用的特征列

FEAT_COLS=[SepalLengthCm,SepalWidthCm,PetalLengthCm,PetalWidthCm]

defmain():

主函数

#数据集

iris_data=pd.read_csv(DATA_FILE,index_col=Id)

iris_data[Label]=iris_data[Species].map(SPECIES_LABEL_DICT)

#获取数据集特征

X=iris_data[FEAT_COLS].values

#获取数据

y=iris_data[Label].values

#划分数据集

X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=1/3,random_state=10)

model_dict={kNN:(KNeighborsClassifier(),

{n_neighbors:[5,15,25],

p:[1,2]}),

LogisticRegression:(LogisticRegression(),

{C:[1e-2,1,1e2]}),

SVM:(SVC(),

{C:[1e-2,1,1e2]})

}

formodel_name,(model,model_par