DL4J - есть ли способ ограничить предсказание модели

Question

Я тренировал модель Mnist с DL4J. Когда я использую эту модель в режиме логического вывода:

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 - результат.