如何从分类器预测()输出?

2024-04-18 02:16:38 发布

您现在位置:Python中文网/ 问答频道 /正文

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

我试过以下方法。。。在

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

请帮助我将top_n_predclass_labels组合起来,得到X_test每行的前n个标签?在

如果有什么捷径可以达到同样的效果,那也是值得欢迎的。在


Tags: 方法test文本列表labelstopnp记录