TensorFlowモデルのパラメータを確認する方法は何ですか。
TensorFlowでは、model.summary()メソッドを使用してモデルのパラメータを表示することができます。このメソッドはモデルの構造と各層のパラメータ数を出力します。例を挙げると、以下の通りです:
import tensorflow as tf
# 创建模型
model = tf.keras.Sequential([
tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10, activation='softmax')
])
# 打印模型参数
model.summary()
上記のコードを実行すると、モデルの構造と各レイヤーのパラメータ数が出力されます。例えば:
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
dense (Dense) (None, 64) 50240
_________________________________________________________________
dense_1 (Dense) (None, 64) 4160
_________________________________________________________________
dense_2 (Dense) (None, 10) 650
=================================================================
Total params: 55,050
Trainable params: 55,050
Non-trainable params: 0
_________________________________________________________________
上記の出力では、各レイヤーのパラメータ数がParam#列に表示されます。