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.
The model predicts fuel consumption using the following features:
- CO Emissions
- Engine Size
- Number of Cylinders
- Python 3.x
- PyTorch
- Pandas
- NumPy
- Matplotlib
To set up the project environment, run the following commands:
pip install torch pandas numpy matplotlib
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.