Rnn 神经网络预测返回意外预测 [关闭]

Question

我正在尝试配置一个 RNN 神经网络来预测 5 种不同类型的文本实体。我正在使用下一个配置：

    MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
            .seed(seed)
            .iterations(100)
            .updater(Updater.ADAM)  //To configure: .updater(Adam.builder().beta1(0.9).beta2(0.999).build())
            .regularization(true).l2(1e-5)
            .weightInit(WeightInit.XAVIER)
            .gradientNormalization(GradientNormalization.ClipElementWiseAbsoluteValue).gradientNormalizationThreshold(1.0)
            .learningRate(2e-2)
            .trainingWorkspaceMode(WorkspaceMode.SEPARATE).inferenceWorkspaceMode(WorkspaceMode.SEPARATE)   //https://deeplearning4j.org/workspaces
            .list()
            .layer(0, new GravesLSTM.Builder().nIn(500).nOut(3)
                    .activation(Activation.TANH).build())
            .layer(1, new RnnOutputLayer.Builder(LossFunctions.LossFunction.MCXENT).activation(Activation.SOFTMAX)        //MCXENT + softmax for classification
                    .nIn(3).nOut(5).build())
            .pretrain(false).backprop(true).build();
  MultiLayerNetwork net = new MultiLayerNetwork(conf);
  net.init();

我训练它，然后评估它。有用。尽管如此，当我使用时：

 int[] prediction = net.predict(features);

有时它会返回和意外预测。它返回正确的预测为 1,2....5，但有时它返回数字为 9,14,12...此数字不对应于可识别的预测/标签。

为什么这个配置会返回意外的输出？

Answer 1

不要使用 net.predict。使用 net.output 和 Nd4j.argMax(outputOfNeuralNet,-1);不应使用 Net.predict（它主要与 2d 一起使用）。