多标签二值化器 - 进行逆变换

Question

我坚持使用 scikit-learn 中的多标签二值化器和一对多分类器。我的挑战是一旦我获得了预测，获取原始标签。（我分别训练和腌制了 one-vs-rest 分类器和向量化器）

_labels = load_labels()
mlb = MultiLabelBinarizer()
mlb.fit_transform(_labels)
print mlb.classes_ # this prints the binarized labels

_clf,_vect = load_pickle('./pickles')

for q in queries:
    #query vector q
    X = vect.transform([q])            
    res = clf.predict_proba(X)
    print res #[[ 0.00164113  0.00706595  0.00683465 .... 0.00837984]]

    #this is where I am stuck on what to pass into the inverse_transform to obtain
    preds = mlb.inverse_transform(??)
    print preds

提前感谢您的帮助！

Answer 1

mlb.fit_transform(_labels) 的输出将是 inverse_transform 的输入。

更多信息在这里：Multilabel Binarizer

多标签二值化器 - 进行逆变换

Multilabel binarizer - getting the inverse transform

python

machine-learning

svm

scipy

scikit-learn