温馨提示×

Scikit-learn中怎么实现多类别分类

小亿
118
2024-05-10 18:34:03
栏目: 编程语言

在Scikit-learn中,可以使用多种算法来实现多类别分类,其中最常用的是使用One-vs-One(OvO)或One-vs-All(OvA)策略。以下是使用Logistic Regression模型进行多类别分类的简单示例:

from sklearn.linear_model import LogisticRegression
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 生成样本数据
X, y = make_classification(n_samples=1000, n_features=20, n_classes=3, n_clusters_per_class=1, random_state=42)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 训练Logistic Regression模型
model = LogisticRegression()
model.fit(X_train, y_train)

# 预测
y_pred = model.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy: ", accuracy)

在上面的示例中,我们使用Logistic Regression模型对3个类别的样本数据进行分类,并计算模型的准确率。实际上,Scikit-learn中的大多数分类算法都支持多类别分类,你可以根据自己的需求选择合适的算法进行实现。

0