tensorflow模型参数如何查看

   2024-10-26 6650
核心提示:在TensorFlow中,可以通过调用model.summary()方法来查看模型的参数。示例如下:import tensorflow as tfmodel = tf.keras.Seque

在TensorFlow中,可以通过调用model.summary()方法来查看模型的参数。示例如下:

import tensorflow as tfmodel = tf.keras.Sequential([    tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)),    tf.keras.layers.Dense(10, activation='softmax')])model.summary()

执行以上代码将会输出模型的参数信息,包括每一层的名称、类型、输出shape以及可训练参数的数量。

 
举报打赏
 
更多>同类网点查询
推荐图文
推荐网点查询
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号