python - 多标签二值化器 - 获取逆变换

Question

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

_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