在我使用tensorflow的第一步中,似乎Model没有训练

时间:2019-02-17 11:51:23

标签: javascript tensorflow

我用JavaScript中的tensorflow迈出了第一步...我认为它应该训练网络,但是渲染只是一条直线...

        var model = tf.sequential();
        model.add(tf.layers.dense({units : 10, inputShape : [1], activation : 'sigmoid'}));
        model.add(tf.layers.dense({units : 10, activation : 'sigmoid'}));
        model.add(tf.layers.dense({units : 1, outputShape : [1], activation : 'sigmoid'}));
        model.compile({optimizer: tf.train.sgd(0.000001),loss: 'meanSquaredError'});
        model.fit(xTrain,yTrain,{epochs : 300,verbose : true});
        var yP = model.predict(xPredict,{epochs : 300,verbose : true});

        window.setInterval(function() {
            render();
        },100);

0 个答案:

没有答案