Question

我正在使用Python进行多标签分类（使用sklearn库），并想知道如何从数组形式的predict_proba（）函数中获取每个预测标签的前5个类，如下所示：

<div style="position:relative;" onclick="thevid=document.getElementById('thevideo');thevid.style.display='block';this.style.opacity='0';this.style.width='0';this.style.height='0';this.style.transition='opacity 1800ms, height 0 1800ms, width 0 1800ms';document.getElementById('iframe').src=document.getElementById('iframe').src.replace('autoplay=0','autoplay=1');">
Click me to hide
</div>
<div id="thevideo" style="display: none;">
  <iframe id="iframe" width="1280" height="720" src="https://www.youtube.com/truncated..." frameborder="0" allowfullscreen></iframe>
</div>

怎么办呢？

Answer 1

您可能需要重现您拥有的代码。这是基于某种假设的解决方案。

如果可以存储

predict_proba（）

作为系列使用，

pd.Series（fit.predict_proba（TESTDATA）[0]）

你可以使用

predict_proba（TESTDATA）.nlargest（N = 5）

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

具有最相关预测的Predict_proba数组

1 个答案: