search

ijyliu / computer-vision-project

Using classical and neural image embeddings and finetuned end-to-end networks to achieve top-tier performance on a vehicle type classification task. Containerized and deployed model as a web app
https://cv-web-app-3m4f2rmfzq-uc.a.run.app/
0 stars 0 forks source link

New XGBoost hyperparameters #74

Open ijyliu opened 2 months ago

ijyliu commented 2 months ago 
hyperparameter_grid =  {
            'n_estimators': [100, 200, 300],
            'learning_rate': [0.1, 0.3, 0.5],
            'max_depth': [2, 3, 5],
            'min_child_weight': [1, 3, 5],
            'k_folds': 5,
    }

make edits to

'n_estimators': [100, 500, 1000, 5000, 10000, 15000, 20000], 'learning_rate': [0.05, 0.1, 0.3, 0.5, 0.75],