Coconut-MNIST / src /config.py
ymlin105's picture
refactor: Reorganize project assets
a7c1baa
raw
history blame contribute delete
649 Bytes
import os
# --- Paths ---
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
DATA_DIR = os.path.join(BASE_DIR, "data")
MODELS_DIR = os.path.join(BASE_DIR, "models")
RESULTS_DIR = os.path.join(BASE_DIR, "assets")
for d in [DATA_DIR, MODELS_DIR, RESULTS_DIR]:
os.makedirs(d, exist_ok=True)
SVD_MODEL_PATH = os.path.join(MODELS_DIR, "svd_10class.pkl")
CNN_MODEL_PATH = os.path.join(MODELS_DIR, "cnn_10class.pth")
FASHION_SVD_PATH = os.path.join(MODELS_DIR, "svd_fashion.pkl") # Placeholder if not exists
FASHION_CNN_PATH = os.path.join(MODELS_DIR, "cnn_fashion.pth")
APP_CACHE_PATH = os.path.join(DATA_DIR, "app_cache.npz")