obesity_rp/config.py

# File and Directory Paths
DATASET_FILE = "datasets/ObesityDataSet_raw_and_data_sinthetic.csv"
MODEL_DIR = "model/"

# Target Variable
TARGET_COLUMN = "NObeyesdad"

# Feature Columns
CATEGORICAL_FEATURES = [
    "Gender",
    "family_history_with_overweight",
    "FAVC",
    "CAEC",
    "SMOKE",
    "SCC",
    "CALC",
    "MTRANS",
]

# Numerical Features
NUMERICAL_FEATURES = ["Age", "Height", "Weight", "FCVC", "NCP", "CH2O", "FAF", "TUE"]

# Model Identifiers
RANDOM_FOREST = "RandomForest"
LIGHTGBM = "LightGBM"
XGBOOST = "XGBoost"
MODEL_CHOICES = [RANDOM_FOREST, LIGHTGBM, XGBOOST]