如何利用代码iris数据集验证逻辑回归算法
时间: 2023-11-26 22:02:31 浏览: 53
使用 UCI 机器学习库中的 iris 数据集来练习 ML
可以使用sklearn库中的LogisticRegression模型来实现逻辑回归算法,具体步骤如下:
1. 加载iris数据集:使用sklearn.datasets.load_iris()函数加载iris数据集。
2. 数据预处理:将数据集分为训练集和测试集,使用sklearn.model_selection.train_test_split()函数。
3. 模型训练:使用sklearn.linear_model.LogisticRegression()函数来创建逻辑回归模型,然后使用fit()函数在训练集上进行训练。
4. 模型评估:使用predict()函数对测试集进行预测,并使用sklearn.metrics.accuracy_score()函数计算模型的准确率。
以下是对应的Python代码示例:
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score
# 加载iris数据集
iris = load_iris()
X, y = iris.data, iris.target
# 数据预处理
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
# 创建逻辑回归模型并训练
model = LogisticRegression()
model.fit(X_train, y_train)
# 对测试集进行预测并计算准确率
y_pred = model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)
希望可以帮助到您。
阅读全文