from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import r2_score
import numpy as np
a数组最少是二维的 因为样本属性不能只通过一个属性去描述
a = np.array([[1,1.5],[2,2.6],[3,3.4],[4,4.6],[5,5.3],[6,6.8]])#数据
b = np.array([1.4,2.8,5.6,6.7,5.4,6.1]) #目标
#训练数据，测试数据，训练目标，测试目标
train_data,test_data,train_target,test_target = train_test_split(a,b,test_size=0.2)
#test_size=0.2 表示将a中80%的数据给train_data,20%给test_data,将b中80%的数据给train_target,20%给test_target
lr = LinearRegression()
lr.fit(train_data,train_target)
c = r2_score(test_target,lr.predict(test_data))
print(c)
万水千山总是情，点个关注行不行 加微信 yyf1752432501 领取python免费资料（python基础语法，js,html,vue,flask，爬虫与反爬虫，数据分析，机器学习，人工智能）
