将文件读取的数据切分,随机选择80%的数据作为训练集,剩余的20%的数据作为测试集
from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(data_regression['registered'], data_regression['cnt'], test_size = 0.20)
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.8, random_state=42)
# 其中 test_size是样本占比,如果是整数的话就是样本的数量;
# random_state是随机数的种子,不同的种子会造成不同的随机采样结果,相同的种子采样结果相同。
望采纳