A JavaScript library for training and deploying ML models in the browser and on Node.js.
Setup Tensorflow.js on Mac
In order to run tensorflow.js in node.js on Mac, you should install tensorflow.js as following:
$ npm install @tensorflow/tfjs-node
If your system NVIDIA GPU, it will be recommended to install below package as well.
$ npm install @tensorflow/tfjs-node-gpu
The below is the example of tensorflow.js based on node.js
const tf = require('@tensorflow/tfjs'); // Load the binding: require('@tensorflow/tfjs-node'); // Use '@tensorflow/tfjs-node-gpu' if running with GPU. // Train a simple model: const model = tf.sequential(); model.add(tf.layers.dense({units: 100, activation: 'relu', inputShape: [10]})); model.add(tf.layers.dense({units: 1, activation: 'linear'})); model.compile({optimizer: 'sgd', loss: 'meanSquaredError'}); const xs = tf.randomNormal([100, 10]); const ys = tf.randomNormal([100, 1]); model.fit(xs, ys, { epochs: 100, callbacks: { onEpochEnd: async (epoch, log) => { console.log(`Epoch ${epoch}: loss = ${log.loss}`); } } });
Reference URL: https://js.tensorflow.org/