code_5.py 216 字节
Newer Older
ToTensor's avatar
ToTensor 已提交
1 2 3
from sklearn.linear_model import Linear Regression #导入线性回归算法模型
model = Linear Regression() #确定线性回归算法
model.fit(X_train, y_train) #根据训练集数据, 训练机器, 拟合函数