Commit
·
449b6cf
1
Parent(s):
4b2a917
add objective function for binary classification
Browse files
main.py
CHANGED
|
@@ -44,6 +44,7 @@ if os.path.exists(MODEL_PATH):
|
|
| 44 |
else:
|
| 45 |
# Hyperparameter tuning
|
| 46 |
param_grid = {
|
|
|
|
| 47 |
"max_depth": [4, 5, 6],
|
| 48 |
"learning_rate": [0.01, 0.05, 0.1],
|
| 49 |
"n_estimators": [100, 200, 300],
|
|
|
|
| 44 |
else:
|
| 45 |
# Hyperparameter tuning
|
| 46 |
param_grid = {
|
| 47 |
+
"objective": ["binary:logistic"], # For binary classification
|
| 48 |
"max_depth": [4, 5, 6],
|
| 49 |
"learning_rate": [0.01, 0.05, 0.1],
|
| 50 |
"n_estimators": [100, 200, 300],
|