File size: 599 Bytes
b8bf9dd
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import os

# Directories
MODEL_DIR = "models"

# Model File Paths
CATBOOST_MODEL_PATH = os.path.join(MODEL_DIR, "catboost_model.cbm")
XGB_MODEL_PATH = os.path.join(MODEL_DIR, "xgb_model.json")
RF_MODEL_PATH = os.path.join(MODEL_DIR, "rf_model.pkl")

# Model Parameters
CATBOOST_PARAMS = {"iterations": 800, "depth": 6, "learning_rate": 0.05, "random_seed": 42, "task_type": "CPU", "verbose": 100}
XGB_PARAMS = {"n_estimators": 800, "learning_rate": 0.05, "max_depth": 6, "tree_method": "hist", "random_state": 42}
RF_PARAMS = {"n_estimators": 200, "max_depth": 15, "random_state": 42, "n_jobs": -1}