linearregression()怎么用
时间: 2023-07-08 12:10:48 浏览: 68
linearregression()是一个函数,通常用于线性回归模型的训练和预测。在使用之前,需要先导入相应的库:
```python
from sklearn.linear_model import LinearRegression
```
接着,我们可以定义一个LinearRegression对象,并使用.fit()方法来拟合训练数据。例如:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
x_train = np.array([[1], [2], [3], [4], [5]])
y_train = np.array([2, 4, 6, 8, 10])
model = LinearRegression()
model.fit(x_train, y_train)
```
上述代码中,我们定义了一个简单的训练集,包含了五个样本。然后,我们创建了一个LinearRegression对象,并使用.fit()方法拟合了训练集。模型现在已经训练好了,可以使用.predict()方法来进行预测。例如:
```python
x_test = np.array([[6], [7], [8], [9], [10]])
y_test = np.array([12, 14, 16, 18, 20])
y_pred = model.predict(x_test)
print(y_pred)
```
上述代码中,我们定义了一个测试集,并使用.predict()方法来预测测试集中的输出。最后,我们可以使用评估指标来评估模型的性能。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)