機器學習5:評估器estimator

要定義與tf.estimator一起使用的自定義模型,需要使用tf.estimator.Estimator。  tf.estimator.LinearRegressor()線性迴歸實際上是一個tf.estimator.Estimator的子類。  我們只是給Estimator提供了一個函數model_fn,它告訴tf.estimator如何評估預測,訓練步驟和損失,而不是分類Estimator。
相關文章
相關標籤/搜索