如何从 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]可能会失败，但随后只需使用迭代器即可。