如何使用 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://tensorflowcn.cn/tutorials/keras/regression

输出

解释

  • DNN 指的是深度神经网络,在本例中它只有一个输入,即“马力”。

  • 此模型适合训练数据。

  • 存储在“history”中的统计参数在控制台上绘制。

  • 进行预测,并使用“evaluate”方法对其进行评估。

更新于: 2021年1月20日

121 次浏览

开启你的 职业生涯

通过完成课程获得认证

立即开始
广告