Skip to content

Commit

Permalink
added one polynomial regression model
Browse files Browse the repository at this point in the history
  • Loading branch information
supremepandey authored Nov 14, 2024
1 parent cfee459 commit 6c3a98f
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion content/07.preliminary_predictive_modeling.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,5 +42,11 @@ To test these theories, we first implemented cross-validation with 5 folds, as t
| MSE | [13,923,954.82, 13,929,781.75, 13,221,049.43, 12,032,485.40, 13,155,213.99] | 13,252,497.08 | 694,039.12 |
| \( R^2 \) | [0.378, 0.369, 0.377, 0.380, 0.379] | 0.377 | 0.004 |

The relatively low performance of linear models suggests that the relationships between the variables in the dataset may be non-linear or involve complex interactions between variables. Therefore, we implemented Decision Trees, Ensemble Methods, and Neural Networks to develop a more suitable model.
The relatively low performance of linear models suggests that the relationships between the variables in the dataset may be non-linear or involve complex interactions between variables. Therefore, we evaluated the performance with the polynomial regression model. This time we eliminated the one feature "Holiday" as it has less correlation with the price and we divided the data into 80% training and 20% testing.

**Equation (5):**
$$\text{Price} = 4332.83 + 4764.5065 \times \text{Total Stops} + 4.98 \times \text{Flight Duration} - 424.07 \times \text{Holiday}<sup>2</sup> - 0.001 \times \text{Flight Duration}<sup>2</sup>$$ - 1.43 \text{Noise}$$

The RSME, MSE and R^2 values we observed from this model are 3082.22, \( 9.50 e+06 \), and 0.48, respectively. You can observed that the model performence have improved than the last model but still it has low performance.


0 comments on commit 6c3a98f

Please sign in to comment.