tensorflojs 如何查看模型结构
在TensorFlow.js中,可以使用model.summary()方法来查看模型的结构。示例如下:
const model = tf.sequential();
model.add(tf.layers.dense({units: 10, inputShape: [784], activation: 'relu'}));
model.add(tf.layers.dense({units: 10, activation: 'softmax'}));
model.summary();
上述代码创建了一个包含两个密集层的序贯模型,并通过model.summary()方法打印出模型的结构信息。输出结果将显示每个层的名称、输出形状和参数数量等信息。
原文地址: http://www.cveoy.top/t/topic/h1Bb 著作权归作者所有。请勿转载和采集!