溫馨提示×

TensorFlow中怎么定義模型結構

小億
119
2024-05-10 15:08:01
欄目: 深度學習

在TensorFlow中,可以使用tf.keras.Sequential()來定義模型結構。Sequential()是一個按照順序構建的模型容器,可以通過添加層來定義模型的結構。

例如,下面是一個簡單的使用Sequential()定義一個包含兩個全連接層的神經網絡模型的示例:

import tensorflow as tf

model = tf.keras.Sequential([
    tf.keras.layers.Flatten(input_shape=(28, 28)),  # 將輸入數據展平為一維向量
    tf.keras.layers.Dense(128, activation='relu'),  # 添加第一個全連接層,128個神經元,激活函數為ReLU
    tf.keras.layers.Dense(10, activation='softmax')  # 添加輸出層,10個神經元,激活函數為softmax
])

在上面的例子中,我們首先通過Sequential()創建一個模型容器,然后添加了一個Flatten層,將輸入數據展平為一維向量,然后添加了一個包含128個神經元的全連接層,激活函數為ReLU,最后添加了一個包含10個神經元的輸出層,激活函數為softmax。

除了使用Sequential()外,還可以通過繼承tf.keras.Model類來自定義模型結構。這種方式更加靈活,可以定義更加復雜的模型結構。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女