Last active
March 23, 2021 23:14
-
-
Save mohdsanadzakirizvi/13a578fe439183b100c1a440e6da19e3 to your computer and use it in GitHub Desktop.
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