import joblib

# These should match what you used in training
feature_params = {
    'max_pad_len': 174,  # From your extract_features() function
    'sample_rate': 22050,  # Common default, adjust if you used different
    'n_mfcc': 40,  # From your MFCC extraction
    'augmentation_config': {  # From your augmentation setup
        'add_noise': True,
        'time_shift': True,
        'pitch_shift': True,
        'time_stretch': True,
        'volume_change': True,
        'background_noise': True,
        'freq_mask': True,
        'time_mask': True
    }
}

joblib.dump(feature_params, "feature_params.joblib")
print("Feature parameters saved.")