如何使用 TensorFlow 在 Auto MPG 資料集上構建 DNN(深度神經網路)模型?


   Tensorflow 是 Google 提供的一個機器學習框架。它是一個開源框架,與 Python 結合使用以實現演算法、深度學習應用程式等等。它用於研究和生產目的。

張量是 TensorFlow 中使用的一種資料結構。它有助於連線流圖中的邊。此流圖稱為“資料流圖”。張量只不過是多維陣列或列表。

我們使用的資料集稱為“Auto MPG”資料集。它包含 1970 年代和 1980 年代汽車的燃油效率。它包括重量、馬力、排量等屬性。有了這些,我們需要預測特定車輛的燃油效率。

我們使用 Google Colaboratory 來執行以下程式碼。Google Colab 或 Colaboratory 幫助在瀏覽器上執行 Python 程式碼,無需任何配置,並可免費訪問 GPU(圖形處理單元)。Colaboratory 構建在 Jupyter Notebook 之上。以下是程式碼片段 -

示例

print("DNN model")
history = dnn_horsepower_model.fit(
   train_features['Horsepower'], train_labels,
   validation_split=0.2,
   verbose=0, epochs=100)
print("Error with respect to every epoch")
plot_loss(history)
x = tf.linspace(0.0, 250, 251)
y = dnn_horsepower_model.predict(x)
plot_horsepower(x, y)
test_results['dnn_horsepower_model'] = dnn_horsepower_model.evaluate(
   test_features['Horsepower'], test_labels,
   verbose=0)

程式碼來源 - https://www.tensorflow.org/tutorials/keras/regression

輸出

解釋

  • DNN 指的是深度神經網路,在本例中它只有一個輸入,即“馬力”。

  • 此模型適合訓練資料。

  • 儲存在“history”中的統計引數在控制檯上繪製。

  • 進行預測,並使用“evaluate”方法對其進行評估。

更新於: 2021年1月20日

121 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

立即開始
廣告