train_model.py
test_size = 0.2
n_estimators = 100
max_depth = 4

Hyperparameters