How can a sequential model be built on Auto MPG dataset using TensorFlow?

TensorFlow is a machine learning framework provided by Google. It is an open-source framework used in conjunction with Python to implement algorithms, deep learning applications and much more. It is used in research and for production purposes.

The 'tensorflow' package can be installed on Windows using the below line of code:

pip install tensorflow

Tensor is a data structure used in TensorFlow. It helps connect edges in a flow diagram known as the 'Data flow graph'. Tensors are multidimensional arrays that can be identified using three main attributes:

  • Rank - It tells about the dimensionality of the tensor or the number of dimensions in the tensor.

  • Type - It tells about the data type associated with the elements of the Tensor.

  • Shape - It is the number of rows and columns together.

Auto MPG Dataset Overview

The aim behind a regression problem is to predict the output of a continuous variable, such as fuel efficiency. The 'Auto MPG' dataset contains fuel efficiency data of 1970s and 1980s automobiles. It includes attributes like weight, horsepower, displacement, and so on. With this, we need to predict the fuel efficiency of specific vehicles.

Building a Sequential Model

A sequential model in TensorFlow is built by stacking layers sequentially. Here's how to create a deep neural network for the Auto MPG regression task:

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

def build_compile_model(norm):
    model = keras.Sequential([
        norm,
        layers.Dense(64, activation='relu'),
        layers.Dense(64, activation='relu'),
        layers.Dense(1)
    ])
    
    model.compile(loss='mean_absolute_error',
                  optimizer=tf.keras.optimizers.Adam(0.001))
    return model

# Assuming horsepower_normalizer is already defined
print("The model is being built and compiled")
dnn_horsepower_model = build_compile_model(horsepower_normalizer)
print("The statistical summary is being computed")
dnn_horsepower_model.summary()

Model Architecture Explanation

  • The model starts with a normalization layer that standardizes input features

  • Two hidden layers with 64 neurons each use ReLU activation functions

  • The output layer has 1 neuron for regression prediction (no activation function)

  • Mean Absolute Error (MAE) is used as the loss function for regression

  • Adam optimizer with learning rate 0.001 is used for training

Model Summary Output

The summary() method displays the model architecture, showing layer types, output shapes, and trainable parameters. This helps understand the model structure and complexity.

Normalization Dense (64, ReLU) Dense (64, ReLU) Dense (1) Input: Horsepower Hidden Layer 1 Hidden Layer 2 Output: MPG Prediction

Key Features

  • Sequential Architecture - Layers are stacked linearly one after another

  • Feature Normalization - Input features are standardized for better training

  • Deep Network - Two hidden layers allow the model to learn complex patterns

  • Regression Output - Single output neuron predicts continuous MPG values

Conclusion

This sequential model architecture with normalization and dense layers provides an effective approach for regression tasks. The model can learn complex relationships between vehicle attributes and fuel efficiency through its deep neural network structure.

Updated on: 2026-03-25T15:39:19+05:30

278 Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements