config.py

# Model configuration
MODEL_CONFIG = {
    'latent_dim': 32,
    'nepochs': 1000,
    'bsize': 16,
    'loss_rec_mult': 100,
    'loss_decor_mult': 10,
    'lr': 1e-4
}

# Preprocessing configuration
PREPROCESS_CONFIG = {
    't_r': 2.0,
    'high_pass': 0.01,
    'low_pass': 0.1,
    'radius': 5
}

# Dataset configuration
DATASET_CONFIG = {
    'name': 'SreekarB/OSFData',
    'split': 'train'
}

# Prediction configuration
PREDICTION_CONFIG = {
    'n_estimators': 100,
    'max_depth': None,
    'cv_folds': 5,
    'prediction_type': 'regression',
    'default_outcome': 'wab_aq',
    'save_path': 'results/treatment_predictor.joblib'
}