Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | U | V _ __getitem__() (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) A apply_namespace() (torch_brain.dataset.Dataset method) B bin_spikes() (in module torch_brain.utils) BINARY (torch_brain.registry.DataType attribute) BinSpikes (class in torch_brain.transforms) C CalciumImagingDatasetMixin (class in torch_brain.dataset) CalciumPOYOPlus (class in torch_brain.models) chain() (in module torch_brain.data) collate() (in module torch_brain.data) Compose (class in torch_brain.transforms) compute_average_firing_rates() (torch_brain.dataset.SpikingDatasetMixin method) ConditionalChoice (class in torch_brain.transforms) CONTINUOUS (torch_brain.registry.DataType attribute) create_linspace_latent_tokens() (in module torch_brain.utils) create_start_end_unit_tokens() (in module torch_brain.utils) CrossEntropyLoss (class in torch_brain.nn) D Dataset (class in torch_brain.data) (class in torch_brain.dataset) DatasetIndex (class in torch_brain.dataset) datasets (torch_brain.dataset.NestedDataset property) DataType (class in torch_brain.registry) detokenizer() (torch_brain.nn.InfiniteVocabEmbedding method) dim (torch_brain.registry.ModalitySpec attribute) disable_data_leakage_check() (torch_brain.data.Dataset method) DistributedEvaluationSamplerWrapper (class in torch_brain.data.sampler) DistributedStitchingFixedWindowSampler (class in torch_brain.data.sampler) E Embedding (class in torch_brain.nn) extend_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) extra_repr() (torch_brain.nn.InfiniteVocabEmbedding method) F FeedForward (class in torch_brain.nn) forward() (torch_brain.models.CalciumPOYOPlus method) (torch_brain.models.POYO method) (torch_brain.models.POYOPlus method) (torch_brain.nn.CrossEntropyLoss method) (torch_brain.nn.FeedForward method) (torch_brain.nn.InfiniteVocabEmbedding method) (torch_brain.nn.Loss method) (torch_brain.nn.MallowDistanceLoss method) (torch_brain.nn.MSELoss method) (torch_brain.nn.MultitaskReadout method) (torch_brain.nn.RotaryCrossAttention method) (torch_brain.nn.RotarySelfAttention method) (torch_brain.nn.RotaryTimeEmbedding method) (torch_brain.nn.SinusoidalTimeEmbedding method) forward_varlen() (torch_brain.nn.MultitaskReadout method) (torch_brain.nn.RotaryCrossAttention method) (torch_brain.nn.RotarySelfAttention method) G get() (torch_brain.data.Dataset method) get_brainset_ids() (torch_brain.data.Dataset method) get_channel_ids() (torch_brain.dataset.MultiChannelDatasetMixin method) get_modality_by_id() (in module torch_brain.registry) get_recording() (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) get_recording_config_dict() (torch_brain.data.Dataset method) get_recording_data() (torch_brain.data.Dataset method) get_recording_hook() (torch_brain.dataset.Dataset method) get_roi_ids() (torch_brain.dataset.CalciumImagingDatasetMixin method) get_sampling_intervals() (torch_brain.data.Dataset method) (torch_brain.dataset.Dataset method) (torch_brain.dataset.NestedDataset method) get_session_ids() (torch_brain.data.Dataset method) get_subject_ids() (torch_brain.data.Dataset method) get_unit_ids() (torch_brain.data.Dataset method) (torch_brain.dataset.SpikingDatasetMixin method) I id (torch_brain.registry.ModalitySpec attribute) InfiniteVocabEmbedding (class in torch_brain.nn) initialize_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) invert() (torch_brain.nn.RotaryTimeEmbedding static method) is_lazy() (torch_brain.nn.InfiniteVocabEmbedding method) isin_interval() (in module torch_brain.utils) L load_pretrained() (torch_brain.models.POYO class method) Loss (class in torch_brain.nn) loss_fn (torch_brain.registry.ModalitySpec attribute) M MallowDistanceLoss (class in torch_brain.nn) ModalitySpec (class in torch_brain.registry) module torch_brain.data torch_brain.data.sampler torch_brain.dataset torch_brain.models torch_brain.nn torch_brain.registry torch_brain.transforms torch_brain.utils MSELoss (class in torch_brain.nn) MultiChannelDatasetMixin (class in torch_brain.dataset) MULTILABEL (torch_brain.registry.DataType attribute) MULTINOMIAL (torch_brain.registry.DataType attribute) MultitaskReadout (class in torch_brain.nn) N NestedDataset (class in torch_brain.dataset) np_string_prefix() (in module torch_brain.utils) P pad() (in module torch_brain.data) pad2d() (in module torch_brain.data) pad2d8() (in module torch_brain.data) pad8() (in module torch_brain.data) POYO (class in torch_brain.models) poyo_mp() (in module torch_brain.models) POYOPlus (class in torch_brain.models) prepare_for_multitask_readout() (in module torch_brain.nn) prepare_for_readout() (in module torch_brain.utils) R RandomChoice (class in torch_brain.transforms) RandomCrop (class in torch_brain.transforms) RandomFixedWindowSampler (class in torch_brain.data.sampler) RandomOutputSampler (class in torch_brain.transforms) RandomTimeScaling (class in torch_brain.transforms) rank_len() (torch_brain.data.sampler.DistributedEvaluationSamplerWrapper method) recording_ids (torch_brain.dataset.Dataset property) register_modality() (in module torch_brain.registry) reset_parameters() (torch_brain.nn.Embedding method) (torch_brain.nn.InfiniteVocabEmbedding method) resolve_weights_based_on_interval_membership() (in module torch_brain.utils) RotaryCrossAttention (class in torch_brain.nn) RotarySelfAttention (class in torch_brain.nn) RotaryTimeEmbedding (class in torch_brain.nn) rotate() (torch_brain.nn.RotaryTimeEmbedding static method) S seed_everything() (in module torch_brain.utils) SequentialFixedWindowSampler (class in torch_brain.data.sampler) set_epoch() (torch_brain.data.sampler.DistributedStitchingFixedWindowSampler method) SinusoidalTimeEmbedding (class in torch_brain.nn) SpikingDatasetMixin (class in torch_brain.dataset) stitch() (in module torch_brain.utils) subset_vocab() (torch_brain.nn.InfiniteVocabEmbedding method) T timestamp_key (torch_brain.registry.ModalitySpec attribute) tokenize() (torch_brain.models.CalciumPOYOPlus method) (torch_brain.models.POYO method) (torch_brain.models.POYOPlus method) tokenizer() (torch_brain.nn.InfiniteVocabEmbedding method) torch_brain.data module torch_brain.data.sampler module torch_brain.dataset module torch_brain.models module torch_brain.nn module torch_brain.registry module torch_brain.transforms module torch_brain.utils module track_batch() (in module torch_brain.data) track_mask() (in module torch_brain.data) track_mask2d() (in module torch_brain.data) track_mask2d8() (in module torch_brain.data) track_mask8() (in module torch_brain.data) TrialSampler (class in torch_brain.data.sampler) TriangleDistribution (class in torch_brain.transforms) type (torch_brain.registry.ModalitySpec attribute) U UnitDropout (class in torch_brain.transforms) UnitFilter (class in torch_brain.transforms) UnitFilterById (class in torch_brain.transforms) V value_key (torch_brain.registry.ModalitySpec attribute)
