Tensorflow.js is an open-source library that is developed by Google for running machine learning models as well as deep learning neural networks in the browser or node environment.
The .train.sgd() function is used to build a tf.SGDOptimizer which utilizes stochastic gradient descent.
Syntax:
tf.train.sgd(learningRate)
Parameters:
- learningRate: The stated learning rate which is used in favor of the SGD algorithm. It is of type number.
Return Value: It returns tf.SGDOptimizer.
Â
Example 1:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs"Â Â // Declaring learningRate const learning_rate = 0.02; Â Â // Calling train.sgd() method const res = tf.train.sgd(learning_rate); Â Â // Printing output console.log(res); |
Output:
{
"learningRate": 0.02,
"c": {
"kept": true,
"isDisposedInternal": false,
"shape": [],
"dtype": "float32",
"size": 1,
"strides": [],
"dataId": {
"id": 1298
},
"id": 1192,
"rankType": "0",
"scopeId": 1251
}
}
Example 2:
Javascript
// Importing the tensorflow.js library import * as tf from "@tensorflow/tfjs"Â Â // Calling train.sgd() and sin() method console.log(JSON.stringify(tf.train.sgd(tf.sin(45)))); |
Output:
{"learningRate":{"kept":false,"isDisposedInternal":false,"shape":[],
"dtype":"float32","size":1,"strides":[],"dataId":{"id":1316},"id":1207,
"rankType":"0","scopeId":1269},"c":{"kept":true,"isDisposedInternal":false,
"shape":[],"dtype":"float32","size":1,"strides":[],"dataId":{"id":1318},
"id":1208,"rankType":"0","scopeId":1269}}
Reference: https://js.tensorflow.org/api/latest/#train.sgd
Whether you’re preparing for your first job interview or aiming to upskill in this ever-evolving tech landscape, neveropen Courses are your key to success. We provide top-quality content at affordable prices, all geared towards accelerating your growth in a time-bound manner. Join the millions we’ve already empowered, and we’re here to do the same for you. Don’t miss out – check it out now!
[…] TensorFlow.js tf.train.sgd() Function […]