我试图得到一个基于文本的多标签分类问题的前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_pred
和class_labels
组合起来,得到X_test
每行的前n个标签?在
如果有什么捷径可以达到同样的效果,那也是值得欢迎的。在
Tags:
我将首先尝试
class_labels[top_n_pred]
,这可能会失败,但只需使用迭代器。在相关问题 更多 >
编程相关推荐