python - 如何从分类器.predict_proba() 输出中获取前 n 个预测标签？

Question

我试图将基于文本的多标签分类问题的前 n 个预测标签作为特定记录的列表。

我尝试了以下...

y_pred_proba = classifier.predict_proba(X_test) 
n = 5
top_n_pred = np.argsort(y_pred_proba, axis=1)[:,-n :]
class_labels = classifier.classes_

请帮助我合并top_n_pred并class_labels获取前 n 个标签作为每行的列表X_test？

如果有任何捷径可以实现相同的目标，那也是受欢迎的。

标签： pythonmachine-learningscikit-learnmulticlass-classification

Answer 1

我会首先尝试class_labels[top_n_pred]可能会失败，然后使用迭代器来完成。