Skip to content

Files

Latest commit

 

History

History
30 lines (25 loc) · 987 Bytes

README.md

File metadata and controls

30 lines (25 loc) · 987 Bytes

Fuel Consumption Prediction Project

Overview

This project involves building and training a machine learning linear regression model to predict fuel consumption based on various vehicle characteristics. The model is built using PyTorch. Dataset derives from kaggle.

Features

The model predicts fuel consumption using the following features:

  • CO Emissions
  • Engine Size
  • Number of Cylinders

Requirements

  • Python 3.x
  • PyTorch
  • Pandas
  • NumPy
  • Matplotlib

Installation

To set up the project environment, run the following commands:

pip install torch pandas numpy matplotlib

Model Training

The training process involves the following steps:

  • Splitting the dataset into training and testing sets.
  • Normalizing the feature variables.
  • Training the model using the Mean Squared Error (MSE) loss function.
  • Using an SGD optimizer with a learning rate of 0.01 and l1 regularization.