机器学习逻辑回归模型鸢尾花数据集
时间: 2023-11-15 21:07:11 浏览: 222
逻辑回归是一种广泛应用于分类问题的机器学习算法。鸢尾花数据集是一个经典的分类问题数据集,它包含了三种不同种类的鸢尾花,每种鸢尾花有四个特征:花萼长度、花萼宽度、花瓣长度和花瓣宽度。逻辑回归模型可以用来对鸢尾花进行分类,根据鸢尾花的四个特征来预测鸢尾花的种类。逻辑回归模型的创建和应用过程与线性回归模型类似,但是需要进行结果分类。具体步骤如下:
1. 导入模型函数:from sklearn.linear_model import LogisticRegression
2. 进行模型初始化:model = LogisticRegression()
3. 模型训练:model.fit(X_train,y_train)
4. 模型得分/评估:model.score(X_test,y_test)
5. 模型预测:model.predict(X_test)
在鸢尾花数据集上,我们可以使用逻辑回归模型来对鸢尾花进行分类预测。具体步骤如下:
1. 导入数据集:from sklearn.datasets import load_iris
2. 加载数据集:iris = load_iris()
3. 划分数据集:from sklearn.model_selection import train_test_split X_train,X_test,y_train,y_test = train_test_split(iris.data,iris.target,test_size=0.3)
4. 创建模型:from sklearn.linear_model import LogisticRegression model = LogisticRegression()
5. 训练模型:model.fit(X_train,y_train)
6. 预测结果:y_pred = model.predict(X_test)
7. 评估模型:from sklearn.metrics import accuracy_score accuracy_score(y_test,y_pred)
阅读全文