config.py

import os

# --- Paths ---
DATA_PATH = './data/synthetic_transactions_samples_5000.csv'
LABEL_ENCODERS_PATH = './label_encoders.pkl'
MODEL_SAVE_DIR = './saved_models/'
PREDICTIONS_SAVE_DIR = './predictions/'  # For storing ensemble results or predictions
TFIDF_VECTORIZER_PATH = os.path.join(MODEL_SAVE_DIR, "tfidf_vectorizer.pkl")

# --- Data Columns ---
TEXT_COLUMN = "Sanction_Context"

LABEL_COLUMNS = [
    "Red_Flag_Reason",
    "Maker_Action",
    "Escalation_Level",
    "Risk_Category",
    "Risk_Drivers",
    "Investigation_Outcome"
]

# Optional Metadata Columns
METADATA_COLUMNS = []  # Add if needed

# --- TF-IDF Settings ---
TFIDF_MAX_FEATURES = 5000
BATCH_SIZE = 16
NUM_EPOCHS = 1