【发布时间】:2016-03-20 10:22:10
【问题描述】:
我正在尝试使用 sci-kit learn 0.17 进行多标签分类 我的数据看起来像
训练
Col1 Col2
asd dfgfg [1,2,3]
poioi oiopiop [4]
测试
Col1
asdas gwergwger
rgrgh hrhrh
到目前为止我的代码
import numpy as np
from sklearn import svm, datasets
from sklearn.metrics import precision_recall_curve
from sklearn.metrics import average_precision_score
from sklearn.cross_validation import train_test_split
from sklearn.preprocessing import label_binarize
from sklearn.multiclass import OneVsRestClassifier
def getLabels():
traindf = pickle.load(open("train.pkl","rb"))
X = traindf['Col1']
y = traindf['Col2']
# Binarize the output
from sklearn.preprocessing import MultiLabelBinarizer
y=MultiLabelBinarizer().fit_transform(y)
random_state = np.random.RandomState(0)
# Split into training and test
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=.5,
random_state=random_state)
# Run classifier
from sklearn import svm, datasets
classifier = OneVsRestClassifier(svm.SVC(kernel='linear', probability=True,
random_state=random_state))
y_score = classifier.fit(X_train, y_train).decision_function(X_test)
但现在我明白了
ValueError: could not convert string to float: <value of Col1 here>
开
y_score = classifier.fit(X_train, y_train).decision_function(X_test)
我也必须对 X 进行二值化吗?为什么我需要将 X 维度转换为浮点数?
【问题讨论】:
-
“您似乎在使用旧的多标签数据表示。不再支持序列序列;请改用二进制数组或稀疏矩阵。” - 你看到了吗?
-
如何将标签转换为二进制数组?
-
this不是同一个问题吗?
-
差不多,但我这里没有使用 LogisticRegression
标签: python machine-learning scikit-learn multilabel-classification