DL4J - 有没有办法限制模型的预测
DL4J - Is there a way to restrict the prediction of a model
我用 DL4J 训练了一个 Mnist 模型。当我在推理模式下使用这个模型时：
INDArray prediction = myModel.output(myINDArrayImage);
这给了我一个 INDArray 的预测，它工作正常。
INDArray 的大小等于我的 OutputLayer 模型的输出数量。
有没有办法限制对字符库的预测？
即像这样的东西：
INDArray prediction = myModel.output(myINDArrayImage, charactersPossible);
其中 charactersPossible 是可能的输出索引列表？
您可以创建一个 INDArray（使用 Nd4j.create(double[])），其中 1.0 表示可能的字符，0.0 表示不可能的字符。然后将其与预测 INDArray 相乘，然后 Nd4j.argMax 结果。
我用 DL4J 训练了一个 Mnist 模型。当我在推理模式下使用这个模型时：
INDArray prediction = myModel.output(myINDArrayImage);
这给了我一个 INDArray 的预测，它工作正常。
INDArray 的大小等于我的 OutputLayer 模型的输出数量。
有没有办法限制对字符库的预测？ 即像这样的东西：
INDArray prediction = myModel.output(myINDArrayImage, charactersPossible);
其中 charactersPossible 是可能的输出索引列表？
您可以创建一个 INDArray（使用 Nd4j.create(double[])），其中 1.0 表示可能的字符，0.0 表示不可能的字符。然后将其与预测 INDArray 相乘，然后 Nd4j.argMax 结果。