请给出分析鸢尾花数据集线性模型的python代码
时间: 2024-04-29 17:25:22 浏览: 125
鸢尾花线性回归+鸢尾花数据集 Python实现
下面是分析鸢尾花数据集线性模型的 Python 代码:
```
from sklearn.datasets import load_iris
from sklearn.linear_model import LinearRegression
# 加载数据集
iris = load_iris()
X = iris.data[:, :2] # 只选取前两个特征
y = iris.target
# 训练线性回归模型
lr = LinearRegression()
lr.fit(X, y)
# 输出模型的系数和截距
print("Coefficients:", lr.coef_)
print("Intercept:", lr.intercept_)
```
以上代码使用 scikit-learn 库中的 `load_iris` 函数加载鸢尾花数据集,并且只选取前两个特征。然后使用 `LinearRegression` 类训练线性回归模型,最后输出模型的系数和截距。
阅读全文