如何使用 TensorFlow 将 DNN(深度神经网络)模型用于预测 Auto MPG 数据集上的 MPG 值?

pythonserver side programmingprogrammingtensorflow

Tensorflow 是 Google 提供的机器学习框架。它是一个与 Python 结合使用的开源框架,用于实现算法、深度学习应用程序等。它用于研究和生产目的。Tensor 是 TensorFlow 中使用的数据结构。它有助于连接流程图中的边缘。此流程图称为"数据流图"。张量不过是多维数组或列表。

"tensorflow"可以使用以下代码行在 Windows 上安装软件包 −

pip install tensorflow

回归问题的目的是预测连续或离散变量的输出,例如价格、概率、是否下雨等等。

我们使用的数据集称为"Auto MPG"数据集。它包含 20 世纪 70 年代和 80 年代汽车的燃油效率。它包括重量、马力、排量等属性。有了这个,我们需要预测特定车辆的燃油效率。

我们使用 Google Colaboratory 来运行以下代码。Google Colab 或 Colaboratory 有助于在浏览器上运行 Python 代码,并且需要零配置和免费访问 GPU

(图形处理单元)。Colaboratory 是在 Jupyter Notebook 之上构建的。以下是代码片段 −

示例

print("Model is being built and compiled")
dnn_model = build_compile_model(normalizer)
print("The statistical summary is displayed ")
dnn_model.summary()

print("The data is being fit to the model")
history = dnn_model.fit(
   train_features, train_labels,
   validation_split=0.2,
   verbose=0, epochs=100)
print("The error versus epoch is visualized")
plot_loss(history)
print("The predictions are being evaluated")
test_results['dnn_model'] = dnn_model.evaluate(test_features, test_labels, verbose=0)

pd.DataFrame(test_results, index=['Mean absolute error [MPG]']).T

代码来源 − https://www.tensorflow.org/tutorials/keras/regression

输出

解释

  • 模型已构建并编译。

  • 使用‘summary’函数显示计数、平均值、中位数等统计值。

  • 此编译模型适合数据。

  • 在控制台上绘制了步骤数与预测误差的可视化。

  • 与线性回归相比,使用 DNN 更好。


相关文章