利用sklearn搭建简单的训练模型

#导入样本数据
from sklearn import datasets
wine = datasets.load_wine()
X = wine.data
y = wine.target

import numpy as np
print(np.shape(X), np.shape(y))

#把数据分成训练数据和测试数据
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 42)
#test_size=0.2  测试数据为总数据的20%
#random_state=42 相当于随机数种子，设置random_state是为了每次分割的训练、测试数据都一样，方便结果复现
print(X_train.shape, X_test.shape)  #print(np.shape(X_train), np.shape(X_test))

#训练模型1：利用逻辑回归模型
print("利用逻辑回归模型来做训练")
from sklearn.linear_model import LogisticRegression
model = LogisticRegression().fit(X_train, y_train) #调用LogisticRegression()里的fit函数进行训练
#https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LinearRegression.html

print("训练数据上的准确率为：%f" % (model.score(X_train, y_train)))
print("测试数据上的准确率为：%f" % (mode