擅长:python、mysql、java
<p>你不需要口述,你可以简单地用zip把你的预测和你的毒物联系起来。然后,您可以使用列表理解来获得1。您可以使用np.where来获取索引值</p>
<pre class="lang-py prettyprint-override"><code>from numpy import array, where
preds = array([1, 0, 1, 0, 1, 1])
toxics = ["Toxic", "Severely Toxic", "Obscene", "Threat", "Inuslt", "Identity Hate"]
results = [toxic for pred, toxic in zip(preds, toxics) if pred == 1]
indexs = where(preds == 1)
print("'The text is", ", ".join(results) + "' as i have a prediction of '1' at", ",".join([str(i) for i in indexs[0]]))
</code></pre>
<p><strong>输出</strong></p>
<pre class="lang-none prettyprint-override"><code>'The text is Toxic, Obscene, Inuslt, Identity Hate' as i have a prediction of '1' at 0,2,4,5
</code></pre>