Prective_proba数组具有最相关的预测



我正在python(使用sklearn库)中进行多标签分类,并想知道如何从drevenct_proba()函数中的每个预测标签中收到前5个类,而阵列则在数组中看起来只是这样:

y_pred=[[1,3,4,5,2],[4,3,2,1,5],[1,2,3,5,4]]

如何完成?

您可能必须重现所拥有的代码。这是基于某些假设的解决方案。

如果您可以存储

predition_proba()

作为系列使用,

pd.Series(fit.predict_proba(testdata)[0])

您可能可以使用

predition_proba(testdata).nlargest(n = 5)

用索引确定前5个概率,自动排序

相关内容

  • 没有找到相关文章

最新更新