肺结节分类算法代码_knn算法对鸢尾花数据集分类（代码分享）

这部分是与bilibili上的视频对应的：

https://www.bilibili.com/video/BV1A7411U7gB/

应有些网友朋友的要求，放一下源码：

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import accuracy_score
import numpy as np

# load data
data = load_iris()
X = data['data']
Y = data['target']

# split arrays or matrices into random train and test subsets
X_train, X_test, Y_train, Y_test= train_test_split(X, Y, test_size=0.2, random_state=0)

# model train
model = KNeighborsClassifier(n_neighbors=3)
model.fit(X_train, Y_train)

# model predict
Y_pred= model.predict(X_test)

# model evaluation
correct_pred = np.count_nonzero(Y_pred==Y_test)
accuracy = correct_pred / len(Y_test)
print("model accuracy is " + str(accuracy))


# print(accuracy_score(Y_test, Y_pred))