| <html> |
| <head> |
| <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.11.2"></script> |
| </head> |
| <body> |
| <div id="output_field"></div> |
| </body> |
| <script> |
| async function learnLinear(){ |
| const model = tf.sequential(); |
| model.add(tf.layers.dense({units: 1, inputShape: [1]})); |
| model.compile({ |
| loss: 'meanSquaredError', |
| optimizer: 'sgd' |
| }); |
| |
| const xs = tf.tensor2d([-1, 0, 1, 2, 3, 4], [6, 1]); |
| const ys = tf.tensor2d([-3, -1, 1, 3, 5, 7], [6, 1]); |
| |
| await model.fit(xs, ys, {epochs: 500}); |
| |
| document.getElementById('output_field').innerText = |
| model.predict(tf.tensor2d([10], [1, 1])); |
| } |
| learnLinear(); |
| </script> |
| <html> |
