Spaces:
Sleeping
Sleeping
File size: 674 Bytes
8cc5633 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# config_streamlit
import os
# Base directory → points to the project root
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
# Model paths
MODEL_PATH_LIGHTGBM = os.path.join(
BASE_DIR, "lightgbm_model", "model", "lightgbm_final_model.pkl"
)
MODEL_PATH_TRANSFORMER = os.path.join(
BASE_DIR, "transformer_model", "model", "checkpoints", "model_final.pth"
)
# Data path
DATA_PATH = os.path.join(
BASE_DIR, "data", "processed", "energy_consumption_aggregated_cleaned.csv"
)
# Color palette for Streamlit layout
PLOT_COLOR = "#edf1f7" # Plot background color
# Constants
TRAIN_RATIO = 0.7 # Train/test split ratio used by both models
|