Skip to content

Model v4

These was the steps of the model-v4 using Random Forest, I also used feature engineering and numerical and categorical transformations.

The model is 21_01_22_lr_w_v3.sav, and can be found on the respective directory.

Pipeline

Transformed Data: second-eda-output.csv
Type conversion
Train Test Split
One Hot Encoder
Fit Random Forest
Predict
Calculate Metrics

Confusion Matrix

confusion-matrix

Feature Importance

feat-importance