1. 导入所需模块和数据集

首先,导入所需的模块,如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)
  1. 训练逻辑回归模型

使用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)
  1. 进行预测及评估

使用训练好的模型对测试集进行预测,并使用相应的评估指标进行模型性能的评估。

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进行逻辑回归分类的基本步骤。

如何使用python进行逻辑回归分类

原文地址: https://www.cveoy.top/t/topic/Br9 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录