Welcome To Ask or Share your Answers For Others

javascript - Get the layers from one model and assign it to another model

posted Oct 24, 2021 in Technique[技术] by 深蓝 (71.8m points)

javascript - Get the layers from one model and assign it to another model

Given a model created using tf.sequential(), is it possible to get the layers and to use them to create another model using tf.model() ?

const model = tf.sequential();
model.add(tf.layers.dense({units: 32, inputShape: [50]}));
model.add(tf.layers.dense({units: 4}));

// get the layers
 layers
// use the layers to create another model
tf.model({layers})

See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙；凝视深渊过久,深渊将回以凝视…

Welcome To Ask or Share your Answers For Others

1 Reply

replyed Oct 24, 2021 by 深蓝 (71.8m points)

To get the layers of the model created using tf.sequential, one needs to use the property layers of the model

const model = tf.sequential();

// first layer
model.add(tf.layers.dense({units: 32, inputShape: [50]}));
// second layer
model.add(tf.layers.dense({units: 4}));

// get all the layers of the model
const layers = model.layers

// second model
const model2 = tf.model({
  inputs: layers[0].input, 
  outputs: layers[1].output
})

model2.predict(tf.randomNormal([1, 50])).print()

<html>
  <head>
    <!-- Load TensorFlow.js -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]"> </script>
  </head>

  <body>
  </body>
</html>

与恶龙缠斗过久,自身亦成为恶龙；凝视深渊过久,深渊将回以凝视…

...