Skip to content

mweglowski/fuel_consumption_prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

8 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

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.

About

πŸš— Linear regression for predicting fuel consumption.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published