DL4J-有没有办法限制模型的预测

Question

我用DL4J训练了Mnist模型。当我在推理模式下使用此模型时：

INDArray prediction = myModel.output(myINDArrayImage);

这给了我INDArray的预测，它工作正常。 INDArray的大小等于我的OutputLayer模型上的输出数量。

有没有一种方法可以将预测限制在字符基础上？ 即类似这样的东西：

INDArray prediction = myModel.output(myINDArrayImage, charactersPossible);

charactersPossible在哪里是可能的输出索引的列表？

Answer 1

您可以创建一个INDArray（使用Nd4j.create（double []）），其中1.0表示可能的字符，而0.0表示不可能的字符。然后将其与预测INDArray相乘，然后将结果与Nd4j.argMax相乘。