Scikit-learn中怎么实现多标签分类-开发者知识库平台

Scikit-learn中怎么实现多标签分类

在Scikit-learn中，可以使用MultiLabelBinarizer来实现多标签分类。以下是一个简单的示例代码：

from sklearn.preprocessing import MultiLabelBinarizer
from sklearn.multiclass import OneVsRestClassifier
from sklearn.svm import SVC

# 定义训练数据
X_train = [[1, 2], [2, 3], [3, 4], [4, 5]]
y_train = [[0, 1], [1, 2], [2], [1, 3]]

# 使用MultiLabelBinarizer将多标签转换为二进制形式
mlb = MultiLabelBinarizer()
y_train_bin = mlb.fit_transform(y_train)

# 定义分类器
clf = OneVsRestClassifier(SVC())

# 训练模型
clf.fit(X_train, y_train_bin)

# 定义测试数据
X_test = [[1, 3], [2, 4]]

# 进行预测
y_pred = clf.predict(X_test)

# 将二进制形式的预测结果转换为多标签形式
y_pred_label = mlb.inverse_transform(y_pred)

print("预测结果：", y_pred_label)

在上面的代码中，首先定义了训练数据X_train和对应的多标签y_train，然后使用MultiLabelBinarizer将多标签转换为二进制形式y_train_bin。接着定义了OneVsRestClassifier分类器，并训练模型。最后定义了测试数据X_test，进行预测，并将预测结果转换为多标签形式输出。

使用MultiLabelBinarizer可以方便地处理多标签分类问题，在训练和预测过程中都能够轻松地转换多标签数据。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

Scikit-learn中怎么实现多标签分类

最新知识库

相关标签