+
80
-

怎么用tensorflow.js来预测价格?

怎么用tensorflow.js来预测价格?

有没有参考教程?

网友回复

+
1
-

当然可以

我来写一个例子

<script type="text/javascript" src="http://repo.bfw.wiki/bfwrepo/js/tf.min.js"></script>
<script type="text/javascript">

    async function predictfuture() {

        ////////////////////////
        // create fake data
        ///////////////////////

        var xs = tf.tensor3d([
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]]
        ]);
        xs.print();

        var ys = tf.tensor3d([
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]],
            [[1], [1], [0]]
        ]);
        ys.print();


        ////////////////////////
        // create model w/ layers api
        ///////////////////////

        console.log('Creating Model...');

        /*

    model design:

                    i(xs)   h       o(ys)
    batch_size  ->  *       *       * -> batch_size
    timesteps   ->  *       *       * -> timesteps
    input_dim   ->  *       *       * -> input_dim


    */

        const model = tf.sequential();

        const hidden = tf.layers.lstm({
            units: 3,
            activation: 'sigmoid',
            inputShape: [3, 1],
            returnSequences: true
        });
        model.add(hidden);

        //output layer
        const output = tf.layers.lstm({
            units: 1,
            activation: 'sigmoid',
            returnSequences: true
        })
        model.add(output);

        //compile
        const sgdoptimizer = tf.train.sgd(0.1)
        model.compile({
            optimizer: sgdoptimizer,
            loss: tf.losses.meanSquaredError
        });

        ////////////////////////
        // train & predict
        ///////////////////////

        console.log('Training Model...');

        await model.fit(xs, ys, {
            epochs: 200
        }).then(() => {

            console.log('Training Complete!');
            console.log('Creating Prediction...');
const inputs = tf.tensor3d( [[[1],[1],[0]]] );
           // const inputs = tf.tensor2d([[1], [1], [0]]);
            let outputs = model.predict(inputs);
            outputs.print();

        });

    }

    predictfuture();

</script>
<h2>tensorflow.js预测未来</h2>
请在console总查看效果

我知道答案,我要回答