本文共 769 字,大约阅读时间需要 2 分钟。
from sklearn import datasetsfrom sklearn.model_selection import train_test_splitfrom sklearn.neighbors import KNeighborsClassifierfrom sklearn.linear_model import LinearRegressionfrom sklearn import treedatas = datasets.load_boston()data_X = datas.datadata_y = datas.target#1:线性回归预测房价model = LinearRegression() model.fit(data_X,data_y) #训练print(model.predict(data_X[:4,:])) #预测print(data_y[:4])iris = datasets.load_iris()iris_X = iris.datairis_y = iris.targetX_train,X_test,y_train,y_test = train_test_split(iris_X,iris_y,test_size=0.3) #训练集测试集划分:留出法、k交叉法、自助法# 2:K近邻分类knn = KNeighborsClassifier()# 3:决策树#dtc = tree.DecisionTreeClassifier()knn.fit(X_train,y_train) #训练print(knn.predict(X_test))#使用模型预测y值print(y_test) # 打印真实标签值y',与y作比对
转载地址:http://enbgi.baihongyu.com/