溫馨提示×

Scikit-learn中怎么實現多類別分類

小億
129
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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女