我正在尝试获取一个基于文本的多标签分类问题的特定记录的前n个预测标签，并将其作为列表返回。

我尝试了以下方法…

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

请帮助我将top_n_pred和class_labels结合起来，以获取X_test的每一行的前n个标签作为列表？

如果有任何快捷方式可以达到同样的效果，也欢迎提供。

回答：

我首先会尝试class_labels[top_n_pred]，这可能会失败，但之后可以用一个迭代器来实现。