HungMingWu · April 30, 2018 07:01
diff --git a/Tensorflow-example.js b/Tensorflow-example.js
 <html>
  <head>
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]"> </script>
    <script>
        const model = tf.sequential();
        model.add(tf.layers.dense({ units: 1, inputShape: [1] }));

        model.compile({ loss: 'meanSquaredError', optimizer: 'sgd' });

        const xs = tf.tensor2d([1, 2, 3, 4], [4, 1]);
        const ys = tf.tensor2d([1, 3, 5, 7], [4, 1]);

        model.fit(xs, ys).then(() => {
            model.predict(tf.tensor2d([5], [1, 1])).print();
        });
    </script>
  </head>

  <body>
  </body>
 </html>
	<html>
	<head>
	<script src="https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]"> </script>
	<script>
	const model = tf.sequential();
	model.add(tf.layers.dense({ units: 1, inputShape: [1] }));

	model.compile({ loss: 'meanSquaredError', optimizer: 'sgd' });

	const xs = tf.tensor2d([1, 2, 3, 4], [4, 1]);
	const ys = tf.tensor2d([1, 3, 5, 7], [4, 1]);

	model.fit(xs, ys).then(() => {
	model.predict(tf.tensor2d([5], [1, 1])).print();
	});
	</script>
	</head>

	<body>
	</body>
	</html>