from sklearn.linear_model import LinearRegression
时间: 2023-07-16 15:16:33 浏览: 212
sklearn常用的API参数解析:sklearn.linear_model.LinearRegression
LinearRegression 是 scikit-learn 库中的一个线性回归模型。可以使用该型对数据进行拟合,并且预测新的数据点。使用时需要导入该模块python
from sklearn.linear_model import LinearRegression
```
然后,通过实例化 LinearRegression 类来创建一个线性回归模型:
```python
model = LinearRegression()
```
接下来,使用 `fit()` 方法对模型进行训练,传入训练数据的自变量和因变量:
```python
model.fit(X_train, y_train)
```
其中,`X_train` 是训练数据的自变量,`y_train` 是训练数据的因变量。
训练完成后,可以使用 `predict()` 方法对新数据进行预测:
```python
y_pred = model.predict(X_test)
```
其中,`X_test` 是新数据的自变量,`y_pred` 是预测的因变量。
除此之外,还可以使用 `score()` 方法计算模型的 R² 分数:
```python
score = model.score(X_test, y_test)
```
其中,`X_test` 是测试数据的自变量,`y_test` 是测试数据的因变量。
阅读全文