gpt4 book ai didi

java - Rnn 神经网络预测返回意想不到的预测

转载 作者:搜寻专家 更新时间:2023-10-31 20:31:57 27 4
gpt4 key购买 nike

<分区>

我正在尝试配置 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...此数字不对应于已识别的预测/标签。

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

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com