Tensorflow.js demo
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
const callbacks = { | |
onEpochEnd: async (epoch, logs) => { | |
console.log("epoch: " + epoch + JSON.stringify(logs)) | |
} | |
}; | |
// Generate some synthetic data for training. | |
const xs = tf.tensor2d([[1], [2], [3], [4]], [4, 1]); | |
const ys = tf.tensor2d([[1], [3], [5], [7]], [4, 1]); | |
// Build and compile model. | |
async function basicRegression(){ | |
// Build a sequential model | |
const model = tf.sequential(); | |
model.add(tf.layers.dense({units: 1, inputShape: [1]})); | |
model.add(tf.layers.dense({units: 1, inputShape: [1]})); | |
model.compile({optimizer: 'sgd', loss: 'meanSquaredError'}); | |
// Train model with fit(). | |
await model.fit(xs, ys, {epochs: 100, validationSplit: 0.1, callbacks: callbacks}); | |
// Run inference with predict(). | |
model.predict(tf.tensor2d([[5]], [1, 1])).print(); | |
} | |
// Create a basic regression model | |
basicRegression(); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<html lang="en"> | |
<head> | |
<meta charset="UTF-8"> | |
<meta name="viewport" content="width=device-width, initial-scale=1.0"> | |
<meta http-equiv="X-UA-Compatible" content="ie=edge"> | |
<!-- load Tensorflow.js --> | |
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@1.0.0/dist/tf.min.js"></script> | |
</head> | |
<body> | |
<h1>Tensorflow.js Core API</h1> | |
<!-- <script src="index.js"></script> --> | |
<script type="text/javascript"> | |
</script> | |
</body> | |
</html> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<html lang="en"> | |
<head> | |
<meta charset="UTF-8"> | |
<meta name="viewport" content="width=device-width, initial-scale=1.0"> | |
<meta http-equiv="X-UA-Compatible" content="ie=edge"> | |
<!-- load Tensorflow.js --> | |
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@1.0.0/dist/tf.min.js"></script> | |
</head> | |
<body> | |
<h1>Tensorflow.js Core API</h1> | |
<!-- <script src="index.js"></script> --> | |
<script type="text/javascript"> | |
const a = tf.tensor([1, 2, 3, 4]); | |
const b = tf.tensor([10, 20, 30, 40]); | |
const y = a.add(b); // equivalent to tf.add(a, b) | |
const z = a.mul(b); // equivalent to tf.mul(a, b) | |
y.print(); | |
z.print(); | |
</script> | |
</body> | |
</html> |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment