Método Tensorflow.js tf.layers getWeights()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node. También ayuda a los desarrolladores a desarrollar modelos ML en lenguaje JavaScript y puede usar ML directamente en el navegador o en Node.js.

La función tf.layers.Layer.getWeights() se utiliza para obtener los valores de los pesos de un tensor.

Sintaxis:

getWeights( trainableOnly? )

Parámetros:

• trainableOnly(boolean): si es verdadero, la función devolverá solo los valores de los pesos que se pueden entrenar.

Valor devuelto: Devuelve un tf.Tensor

Ejemplo 1:

// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units: 2, inputShape: [5]}));
model.add(tf.layers.dense({units: 3}));
  
model.compile({loss: 'categoricalCrossentropy', optimizer: 'sgd'});
  
// Printing the weights of the layers
model.layers[0].getWeights()[0].print()
model.layers[0].getWeights()[1].print()

Producción:

Tensor
    [[-0.4756567, 0.2925433 ],
     [0.3505997 , -0.5043278],
     [0.5344347 , 0.2662918 ],
     [-0.1357223, 0.2435055 ],
     [-0.6059403, 0.1990891 ]]
Tensor
    [0, 0]

Ejemplo 2:

const tf = require("@tensorflow/tfjs")
  
// Creating a model
const model = tf.sequential();
  
// Adding layers
model.add(tf.layers.dense({units: 1, inputShape: [10]}));
model.add(tf.layers.dense({units: 3}));
  
// Setting new weights
model.layers[0].setWeights([tf.zeros([10, 1]), tf.ones([1])]);
  
model.compile({loss: 'categoricalCrossentropy', optimizer: 'sgd'});
  
// Printing the weights of the layers
model.layers[0].getWeights()[0].print()
model.layers[0].getWeights()[1].print()

Producción:

Tensor
    [[0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0],
     [0]]
Tensor
    [1]

Referencia: https://js.tensorflow.org/api/latest/#tf.layers.Layer.getWeights