擅长:python、mysql、java
<p>我也有同样的问题。在</p>
<p>Lgb预测以扁平数组的形式输出。在</p>
<p>通过检查,我发现它是这样的:</p>
<p>样本<code>a</code>到类<code>i</code>的概率位于</p>
<p><code>num_classes*(a-1) + i</code>位置</p>
<p>至于你的代码,应该是这样的:</p>
<pre><code> def evalerror(preds, dtrain):
labels = dtrain.get_label()
preds = preds.reshape(5, -1).T
preds = preds.argmax(axis = 1)
f_score = f1_score(labels , preds, average = 'weighted')
return 'f1_score', f_score, True
</code></pre>