Introduction
構建一個子類的API實現wide & deep模型
- 繼承
tf.keras.models.Model這個class
- 繼承
繼承後除了重載
__init__(self)函數之外,還需重載call(self, input)函數__init__用於定義模型的層次call函數用於完成模型的正向計算
於子類構建完成模型後
實例化子類調用模型
- 實例化子類
- 調用
build函數- 並指定其
input_shape
- 並指定其
使用
tf.keras.models.Sequential類實例化模型- 一樣需要調用
build函數,並指定其input_shape
- 一樣需要調用
Code Presentation