In a regression problem, the aim is to predict the output of a continuous value, like a price or a probability. Contrast this with a classification problem, where the aim is to select a class from a list of classes (for example, where a picture contains an apple or an orange, recognizing which fruit is in the picture).
This tutorial uses the classic Auto MPG dataset and demonstrates how to build models to predict the fuel efficiency of the late-1970s and early 1980s automobiles. To do this, you will provide the models with a description of many automobiles from that time period. This description includes attributes like cylinders, displacement, horsepower, and weight.
2. Preprocessing: mainly cleaning and convert categorical values one hot encoding.
The plot of the HorsePower feature with MPG.
The plot of the HorsePower feature with MPG.
Model | Mean absolute error [MPG] |
---|---|
Single_feature_Regression | 3.644525 |
linear_model_MultipInputs | 3.364622 |
dnn_model_single_input | 2.910135 |
dnn_mi | 2.734736 |
reloaded | 2.734736 |