机器学习之模型训练
- iris数据加载
- 使用scikit-learn建模四步骤
- 1.调用需要使用的模型类
- 2.模型初始化(创建一个模型实例)
- 3.模型训练
- 4.模型预测
from sklearn import datasets
iris = datasets.load_iris()
X = iris.data
y = iris.target
print(X.shape)
print(y.shape)
from sklearn.neighbors import KNeighborsClassifier
knn = KNeighborsClassifier(n_neighbors=1)
print(knn)
knn.fit(X,y)
knn.predict([[1,2,3,4]])
x_test = [[1,2,3,4],[2,4,1,2]]
knn.predict(x_test)
knn_5 = KNeighborsClassifier(n_neighbors=5)
knn_5.fit(X,y)
x_test = [[1,2,3,4],[2,4,1,2]]
knn_5.predict(x_test)
print(knn_5)