机器学习模型训练范本

一，概述

1. 首先是数据获取；
2. 处理训练数据和测试数据；
3. 模型训练；
4. 使用训练好的模型进行预测；

二，源代码分析

from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
from sklearn.datasets import load_iris
from sklearn import datasets

iris=load_iris()
iris_X=iris.data
iris_Y=iris.target
X_train,X_test,Y_train,Y_test=train_test_split(iris_X,iris_Y,test_size=0.3)
lr=LinearRegression()
lr.fit(X_train,Y_train)
result=lr.predict(X_test)


print 'predict the y values :'
print result
print 'the test`s y values ::'
print Y_test