如何使用python进行逻辑回归分类
- 导入所需模块和数据集
首先,导入所需的模块,如numpy、pandas、sklearn等。然后,加载数据集并将其划分为训练集和测试集。
import numpy as np
import pandas as pd
from sklearn import datasets
from sklearn.model_selection import train_test_split
# 加载数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 划分数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
- 训练逻辑回归模型
使用sklearn库中的逻辑回归算法进行训练。在此之前,需要对数据进行预处理,例如特征缩放,以确保所有特征都以相同的比例进行计算。
from sklearn.linear_model import LogisticRegression
from sklearn.preprocessing import StandardScaler
# 特征缩放
scaler = StandardScaler()
X_train_scaled = scaler.fit_transform(X_train)
X_test_scaled = scaler.transform(X_test)
# 训练模型
logreg = LogisticRegression()
logreg.fit(X_train_scaled, y_train)
- 进行预测及评估
使用训练好的模型对测试集进行预测,并使用相应的评估指标进行模型性能的评估。
from sklearn.metrics import accuracy_score, confusion_matrix
# 进行预测
y_pred = logreg.predict(X_test_scaled)
# 评估模型
print("Accuracy: ", accuracy_score(y_test, y_pred))
print("Confusion Matrix: ")
print(confusion_matrix(y_test, y_pred))
以上就是使用Python进行逻辑回归分类的基本步骤。
原文地址: https://www.cveoy.top/t/topic/Br9 著作权归作者所有。请勿转载和采集!