deepextractor.evaluation.simulate

Simulated evaluation of DeepExtractor models.

Usage:

deepextractor-evaluate --model DeepExtractor_257 --checkpoint-dir checkpoints/ \
    --data-dir data/ --output-dir evaluation/

Module Contents

deepextractor.evaluation.simulate.logger

deepextractor.evaluation.simulate.SAMPLE_RATE = 4096

deepextractor.evaluation.simulate.NYQUIST_FREQ = 2048

deepextractor.evaluation.simulate.T = 2.0

deepextractor.evaluation.simulate.T_INJ = 1.0

deepextractor.evaluation.simulate.LENGTH = 0

deepextractor.evaluation.simulate.MODEL_REGISTRY

deepextractor.evaluation.simulate.prepare_data_for_stft(data, scaler, n_fft, hop_length, win_length, window)

deepextractor.evaluation.simulate.calculate_match(true_signal, predicted_signal, sample_rate=SAMPLE_RATE)

deepextractor.evaluation.simulate.generate_glitch_data(signal_type, gaussian_noise_samples, signal_function_map, snr_min=7.5, snr_max=100, bilby_noise=False)

deepextractor.evaluation.simulate.generate_hybrid_glitch_data(gaussian_noise_samples, signal_function_map, snr_min=7.5, snr_max=100)

deepextractor.evaluation.simulate.evaluate_model(model_name, model_registry, scaler, glitch_data, output_path, checkpoint_dir, device, batch_size=8, checkpoint_filename=CHECKPOINT_BILBY)

deepextractor.evaluation.simulate.main()