如何使用 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”方法对其进行评估。
广告