deepextractor.evaluation.simulate ================================= .. py:module:: deepextractor.evaluation.simulate .. autoapi-nested-parse:: Simulated evaluation of DeepExtractor models. Usage:: deepextractor-evaluate --model DeepExtractor_257 --checkpoint-dir checkpoints/ \ --data-dir data/ --output-dir evaluation/ Module Contents --------------- .. py:data:: logger .. py:data:: SAMPLE_RATE :value: 4096 .. py:data:: NYQUIST_FREQ :value: 2048 .. py:data:: T :value: 2.0 .. py:data:: T_INJ :value: 1.0 .. py:data:: LENGTH :value: 0 .. py:data:: MODEL_REGISTRY .. py:function:: prepare_data_for_stft(data, scaler, n_fft, hop_length, win_length, window) .. py:function:: calculate_match(true_signal, predicted_signal, sample_rate=SAMPLE_RATE) .. py:function:: generate_glitch_data(signal_type, gaussian_noise_samples, signal_function_map, snr_min=7.5, snr_max=100, bilby_noise=False) .. py:function:: generate_hybrid_glitch_data(gaussian_noise_samples, signal_function_map, snr_min=7.5, snr_max=100) .. py:function:: evaluate_model(model_name, model_registry, scaler, glitch_data, output_path, checkpoint_dir, device, batch_size=8, checkpoint_filename=CHECKPOINT_BILBY) .. py:function:: main()