我正在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个概率,自动排序