Hike News
Hike News

Tensorflow2-tensorflow-keras-Wide & Deep 房價預測演示(子類API)

Introduction

  • 構建一個子類的API實現wide & deep模型

    • 繼承tf.keras.models.Model這個class
  • 繼承後除了重載__init__(self)函數之外,還需重載call(self, input)函數

    • __init__用於定義模型的層次
    • call函數用於完成模型的正向計算
  • 於子類構建完成模型後

    • 實例化子類調用模型

      1. 實例化子類
      2. 調用build函數
        • 並指定其input_shape
    • 使用tf.keras.models.Sequential類實例化模型

      • 一樣需要調用build函數,並指定其input_shape

Code Presentation