Skip to content
Related Articles

Related Articles

Save Article
Improve Article
Save Article
Like Article

Tensorflow.js tf.LayersModel class .save() Method

  • Last Updated : 01 Sep, 2021

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 .save() function is used to save the structure and/or the weights of the stated LayersModel.

Hey geek! The constant emerging technologies in the world of web development always keeps the excitement for this subject through the roof. But before you tackle the big projects, we suggest you start by learning the basics. Kickstart your web development journey by learning JS concepts with our JavaScript Course. Now at it's lowest price ever!

Note:

  • An IOHandler is an object which possesses a save method concerning the accurate signature specified.
  • The save method controls the accumulation or else transmission of sequential data i.e. artifacts which describes the model’s topology as well as weights upon or by means of a particular medium, like local storagefile, file downloads, IndexedDB in the web browser as well as HTTP requests to a server.
  • TensorFlow.js enables IOHandler implementations in favor of a number of repeatedly utilized saving mediums, like tf.io.browserDownloads() and tf.io.browserLocalStorage.
  • Moreover, this method also permits us to apply specific kinds of IOHandlers such as URL-like string techniques, like ‘localstorage://’ and ‘indexeddb://’.

Syntax:



save(handlerOrURL, config?)

 

Parameters:  

  • handlerOrURL: The stated instance of IOHandler or else a URL resembling, design-based string techniques in favor of IOHandler. It is of type io.IOHandler|string.
  • config: The stated options in order to save the stated model. It is optional and is of type object. It has two arguments under it, as given below:
    1. trainableOnly: It states if only the trainable weights of the stated model is saved, overlooking the non-trainable weights. It is of type Boolean and defaults to false.
    2. includeOptimizer: It states if the stated optimizer will be stored or not. It is of type Boolean and defaults to false.

Return Value: It returns the promise of io.SaveResult.

Example 1:

Javascript




// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential(
     {layers: [tf.layers.dense({units: 2, inputShape: [2]})]});
  
// Calling save() method
const res = await model.save('localstorage://mymodel');
  
// Printing output
console.log(res)

Output:

{
  "modelArtifactsInfo": {
    "dateSaved": "2021-08-23T09:53:28.198Z",
    "modelTopologyType": "JSON",
    "modelTopologyBytes": 612,
    "weightSpecsBytes": 125,
    "weightDataBytes": 24
  }
}

Example 2:

Javascript




// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a model
const model = tf.sequential(
     {layers: [tf.layers.dense({units: 2, inputShape: [2]})]});
  
// Calling save() method
const res = await model.save('localstorage://mymodel', true, true);
  
// Printing output
console.log(JSON.stringify(res))

Output:

{"modelArtifactsInfo":{"dateSaved":"2021-08-23T09:55:33.044Z",
"modelTopologyType":"JSON","modelTopologyBytes":612,
"weightSpecsBytes":125,"weightDataBytes":24}}

Reference: https://js.tensorflow.org/api/latest/#tf.LayersModel.save




My Personal Notes arrow_drop_up
Recommended Articles
Page :