deepSTRF package

Subpackages

• deepSTRF.datasets package
  • Subpackages
    • deepSTRF.datasets.audio package
      • Submodules
      • deepSTRF.datasets.audio.audio_dataset module
      • deepSTRF.datasets.audio.ns1_drc module
      • deepSTRF.datasets.audio.nat4 module
      • deepSTRF.datasets.audio.wehr module
      • deepSTRF.datasets.audio.asari module
      • deepSTRF.datasets.audio.crcns_aa1 module
      • deepSTRF.datasets.audio.crcns_aa2 module
      • deepSTRF.datasets.audio.crcns_aa4 module
      • deepSTRF.datasets.audio.espejo module
      • deepSTRF.datasets.audio.alice_eeg module
      • deepSTRF.datasets.audio.le_2025 module
      • Module contents
    • deepSTRF.datasets.video package
      • Submodules
      • deepSTRF.datasets.video.video_dataset module
      • Module contents
  • Submodules
  • deepSTRF.datasets.neural_dataset module
    • NeuralDataset
      • NeuralDataset.compute_neuron_quality()
      • NeuralDataset.get_N()
      • NeuralDataset.get_S()
      • NeuralDataset.get_nrn_meta()
      • NeuralDataset.normalize_responses()
      • NeuralDataset.nrn_masks
      • NeuralDataset.reset_pop_selection()
      • NeuralDataset.reset_stim_selection()
      • NeuralDataset.select_neuron()
      • NeuralDataset.select_pop_by_nrn_attr()
      • NeuralDataset.select_pop_by_nrn_predicate()
      • NeuralDataset.select_pop_by_stim_attr()
      • NeuralDataset.select_pop_by_stim_predicate()
      • NeuralDataset.select_population()
      • NeuralDataset.select_stim()
      • NeuralDataset.select_stims()
      • NeuralDataset.select_stims_by_attr()
      • NeuralDataset.select_stims_by_predicate()
      • NeuralDataset.smooth_responses()
      • NeuralDataset.standardize_stims()
      • NeuralDataset.validate()
  • Module contents
    • NeuralDataset
      • NeuralDataset.compute_neuron_quality()
      • NeuralDataset.get_N()
      • NeuralDataset.get_S()
      • NeuralDataset.get_nrn_meta()
      • NeuralDataset.normalize_responses()
      • NeuralDataset.nrn_masks
      • NeuralDataset.reset_pop_selection()
      • NeuralDataset.reset_stim_selection()
      • NeuralDataset.select_neuron()
      • NeuralDataset.select_pop_by_nrn_attr()
      • NeuralDataset.select_pop_by_nrn_predicate()
      • NeuralDataset.select_pop_by_stim_attr()
      • NeuralDataset.select_pop_by_stim_predicate()
      • NeuralDataset.select_population()
      • NeuralDataset.select_stim()
      • NeuralDataset.select_stims()
      • NeuralDataset.select_stims_by_attr()
      • NeuralDataset.select_stims_by_predicate()
      • NeuralDataset.smooth_responses()
      • NeuralDataset.standardize_stims()
      • NeuralDataset.validate()
• deepSTRF.metrics package
  • Submodules
  • deepSTRF.metrics.losses module
    • mse_loss()
    • poisson_loss()
  • deepSTRF.metrics.performance module
    • coherence()
    • compute_CCmax()
    • compute_TTRC()
    • corrcoef()
    • fve()
    • noise_power()
    • normalized_corrcoef()
    • signal_power()
    • snr()
  • Module contents
    • coherence()
    • corrcoef()
    • fve()
    • mse_loss()
    • noise_power()
    • normalized_corrcoef()
    • poisson_loss()
    • signal_power()
    • snr()
• deepSTRF.models package
  • Subpackages
    • deepSTRF.models.audio package
      • Submodules
      • deepSTRF.models.audio.audio_model module
      • deepSTRF.models.audio.audio_zoo module
      • Module contents
    • deepSTRF.models.video package
      • Submodules
      • deepSTRF.models.video.video_model module
      • Module contents
  • Submodules
  • deepSTRF.models.layers module
    • CausalLayerNorm
      • CausalLayerNorm.forward()
    • CausalSTRFConv
      • CausalSTRFConv.STRF_weight()
      • CausalSTRFConv.forward()
    • LearnableExponentialDecay
      • LearnableExponentialDecay.build_kernel()
      • LearnableExponentialDecay.forward()
      • LearnableExponentialDecay.tau()
    • LocallyConnected1d
      • LocallyConnected1d.forward()
    • ParametricSTRF
      • ParametricSTRF.build_kernel()
      • ParametricSTRF.forward()
    • SeparableSTRF
      • SeparableSTRF.build_kernel()
      • SeparableSTRF.forward()
    • SinusoidalPositionalEncoding
      • SinusoidalPositionalEncoding.forward()
    • build_causal_window_mask()
  • deepSTRF.models.neural_model module
    • NeuralModel
      • NeuralModel.count_trainable_params()
      • NeuralModel.detach()
      • NeuralModel.forward()
      • NeuralModel.from_pretrained()
      • NeuralModel.push_to_hub()
      • NeuralModel.save_pretrained()
      • NeuralModel.validate()
  • deepSTRF.models.prefiltering module
    • AdapTrans
      • AdapTrans.OFF_kernel()
      • AdapTrans.ON_kernel()
      • AdapTrans.build_kernels()
      • AdapTrans.forward()
      • AdapTrans.get_a()
      • AdapTrans.get_w()
      • AdapTrans.out_channels
      • AdapTrans.plot_kernels()
    • ICAdaptation
      • ICAdaptation.build_kernels()
      • ICAdaptation.forward()
      • ICAdaptation.out_channels
      • ICAdaptation.plot_kernels()
    • Willmore_Adaptation
    • a_to_tau()
    • freq_to_tau()
    • get_CFs()
    • make_prefiltering()
    • tau_to_a()
  • deepSTRF.models.readouts module
    • LinearReadout
      • LinearReadout.forward()
    • STRFReadout
      • STRFReadout.STRF_weight()
      • STRFReadout.forward()
  • deepSTRF.models.activations module
    • ParametricDoubleExponential
      • ParametricDoubleExponential.a
      • ParametricDoubleExponential.b
      • ParametricDoubleExponential.extra_repr()
      • ParametricDoubleExponential.forward()
    • ParametricSigmoid
      • ParametricSigmoid.a
      • ParametricSigmoid.b
      • ParametricSigmoid.extra_repr()
      • ParametricSigmoid.forward()
    • ParametricSoftplus
      • ParametricSoftplus.b
      • ParametricSoftplus.beta
      • ParametricSoftplus.extra_repr()
      • ParametricSoftplus.forward()
  • deepSTRF.models.scales module
    • ERB_bandwidth()
    • ERB_to_Hz()
    • Greenwood()
    • Hz_to_ERB()
    • Hz_to_mel()
    • inverse_Greenwood()
    • mel_to_Hz()
  • Module contents
    • NeuralModel
      • NeuralModel.count_trainable_params()
      • NeuralModel.detach()
      • NeuralModel.forward()
      • NeuralModel.from_pretrained()
      • NeuralModel.push_to_hub()
      • NeuralModel.save_pretrained()
      • NeuralModel.validate()
• deepSTRF.training package
  • Submodules
  • deepSTRF.training.fitter module
    • Fitter
      • Fitter.compute_loss()
      • Fitter.evaluate()
      • Fitter.fit()
      • Fitter.on_epoch_end()
  • deepSTRF.training.multi_seed module
    • fit_multi_seed()
  • deepSTRF.training.config module
    • auto_config()
    • slug_from_config()
  • deepSTRF.training.seed module
    • set_random_seed()
  • deepSTRF.training.wandb_log module
    • WandbSeedLogger
      • WandbSeedLogger.close()
      • WandbSeedLogger.finalize()
    • make_wandb_logger_factory()
  • deepSTRF.training.tb_log module
    • TensorBoardSeedLogger
      • TensorBoardSeedLogger.close()
      • TensorBoardSeedLogger.finalize()
    • make_tensorboard_logger_factory()
  • Module contents
    • Fitter
      • Fitter.compute_loss()
      • Fitter.evaluate()
      • Fitter.fit()
      • Fitter.on_epoch_end()
    • auto_config()
    • fit_multi_seed()
    • set_random_seed()
    • slug_from_config()
• deepSTRF.utils package
  • Submodules
  • deepSTRF.utils.data module
    • ResponseSmoothingTransform
      • ResponseSmoothingTransform.forward()
    • concat_neural_datasets()
    • concatenate_datasets()
    • fill_missing_data()
    • hanning_smooth()
    • neural_collate()
  • deepSTRF.utils.data_download module
    • crcns_download()
    • default_cache_dir()
    • figshare_download()
    • github_raw_download()
    • osf_download()
    • stream_download()
    • untar()
    • unzip()
    • zenodo_download()
  • deepSTRF.utils.plotting module
    • compare_wav2spec_to_groundtruth()
    • plot_psth_vs_pred()
    • plot_stim_with_response()
    • plot_strf_grid()
  • deepSTRF.utils.hub module
    • Public surface
    • download_pretrained()
    • load_pretrained_from_dir()
    • save_pretrained_to_dir()
    • upload_pretrained()
  • Module contents
    • ResponseSmoothingTransform
      • ResponseSmoothingTransform.forward()
    • compare_wav2spec_to_groundtruth()
    • concat_neural_datasets()
    • hanning_smooth()
    • neural_collate()
    • plot_psth_vs_pred()
    • plot_stim_with_response()
    • plot_strf_grid()

Module contents