iT邦幫忙

第 11 屆 iThome 鐵人賽

DAY 16
0
Google Developers Machine Learning

文組腦的機器學習異想世界系列 第 16

【Day16】文組腦簡述Estimator API程式碼

  • 分享至 

  • xImage
  •  

哈囉大家好~~
就如前篇所述,因為Estimator API使用高級又直觀的程式碼開發模型,因此,採用 Estimator API創建模型通常比採用低階的TensorFlow API 更簡單。
在定義特徵時,我們可以使用下面程式碼:

INPUT_COLUMNS = [
    tf.feature_column.numeric_column('…')

我們用下面程式碼來使用線性回歸:

model = tf.estimator.LinearRegressor(
      feature_columns = feature_cols, model_dir = OUTDIR)

我們用下面程式碼來使用深度神經網路,我們建立 3 層的類神經網路,3 層的單元數分別為 32、8、2個:

model = tf.estimator.DNNRegressor(hidden_units = [32, 8, 2],
      feature_columns = make_feature_cols(), model_dir = OUTDIR)

我們使用下面程式碼開始訓練模型:

model.train(input_fn = get_train_input_fn, steps = 200)

如果我們有一組新資料要讓模型做預測,我們可以使用以下程式碼:

predictions = model.predict(input_fn = make_prediction_input_fn(df_test))

以上,程式碼就暫時到這邊!我們明天見~~


上一篇
【Day15】文組腦學習張量(Tensor)、變數(variable) 和Placeholder
下一篇
【Day17】文組腦理介紹雲端的機器學習引擎
系列文
文組腦的機器學習異想世界30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言