java - 如何修复微笑神经网络以进行回归，它总是输出相同的预测

Question

我正在使用微笑库为简单的非线性回归问题训练神经网络。对于每个输入，我都会得到相同的预测。也许我在构建神经网络时做错了什么。你能建议我错过什么吗？由于法律原因，我不包括数据。[1, 3, 8, 15, 7 ...]

    int[] units=new int[4];
    units[0]=1;
    units[1]=10;
    units[2]=10;
    units[3]=1;
    smile.regression.NeuralNetwork net =
        new smile.regression.NeuralNetwork(
         smile.regression.NeuralNetwork.ActivationFunction.LOGISTIC_SIGMOID,
         units);

    int epochs=10;
    for (int i = 0; i < epochs; i++) {
        net.learn(data, label);
    }

    double[] pred = new double[label.length];
    double maxError=0.0;
    for (int i = 0; i < label.length; i++) {
        pred[i] = net.predict(data[i]);
        double error=Math.abs(pred[i]-label[i]);
        if(error>maxError){
            maxError=error;
        }
        System.out.println("For data "+data[i][0]);
        System.out.println(pred[i]+" "+label[i]);
    }

Answer 1

我相信这是由于您的单位向量，它应该只包含 3 个数字：输入层数 = 特征数隐藏层数和输出层数 = 1