机器学习之模型训练

机器学习之模型训练

• iris数据加载
• 使用scikit-learn建模四步骤
• 1.调用需要使用的模型类
• 2.模型初始化（创建一个模型实例）
• 3.模型训练
• 4.模型预测

# 通过sklearn自带数据包加载iris数据
from sklearn import datasets
iris = datasets.load_iris()
#样本数据与结果分别赋值到"X""y"
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)
#设定一个新的K值进行KNN建模
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)