中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

tensorflow模型參數怎么查看

小億
159
2024-01-09 15:24:31
欄目: 深度學習

在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 #這一列中。

0
虞城县| 清远市| 石门县| 浦城县| 丘北县| 道真| 且末县| 曲松县| 灵武市| 名山县| 钟祥市| 张北县| 仪陇县| 五原县| 宁南县| 邢台市| 介休市| 大关县| 稷山县| 巴彦县| 淮阳县| 乐山市| 云阳县| 丁青县| 宁强县| 玉树县| 科技| 藁城市| 桐乡市| 凤凰县| 沁阳市| 长乐市| 新民市| 阿合奇县| 宁南县| 祥云县| 杭州市| 类乌齐县| 融水| 徐闻县| 肥西县|