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