怎么用tensorflow.js来预测价格?
有没有参考教程?
网友回复
当然可以
我来写一个例子
<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总查看效果