如何从 classifier.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?

如果有任何捷径可以达到同样的效果，也欢迎。

Answer 1

我会先尝试 class_labels[top_n_pred]，这可能会失败，但随后就用迭代器来完成。