Index

A

add_common_args() (in module protosleepnet.proto_reconstruction.utils) analyze_one_m() (in module protosleepnet.seed_mmd_analysis) analyze_prototype() (in module protosleepnet.proto_reconstruction.reconstruction_relevance) analyze_task() (in module protosleepnet.clinical_probing.parkinsons.analyze_features) anon_label() (in module protosleepnet.demo.precompute_indomain)

anon_labels() (in module protosleepnet.demo.subset_bundle) ARRAYS (in module protosleepnet.demo.subset_bundle) attention (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute) attn_pool (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) audit_model() (in module protosleepnet.demo.evaluate_explanations)

B

BACKBONE_CONFIGS (in module protosleepnet.train) BAND_FREQ_ORDER (in module protosleepnet.figure_reconstruction.prototype_card) BAND_GREEK (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) BAND_NAMES (in module protosleepnet.demo.evaluate_explanations) BANDS (in module protosleepnet.demo.evaluate_explanations) BASE (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) (in module protosleepnet.proto_reconstruction.utils) BASELINE_PRED_FILES (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) best_per_M() (in module protosleepnet.clinical_probing.parkinsons.generate_tables) build_dataset() (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_residual_prototypes) (in module protosleepnet.test_occlusion) (in module protosleepnet.test_prototypes) (in module protosleepnet.train)

build_dataset_and_loaders() (in module protosleepnet.baselines.extract_residual_embeddings) (in module protosleepnet.extract_proto_embeddings) build_downstream_fn() (in module protosleepnet.learn_prototypes_vq) build_full_loader() (in module protosleepnet.proto_reconstruction.utils) build_model() (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer) (in module protosleepnet.build_protosleepnet) build_prototype_cards() (in module protosleepnet.demo.build_cards) build_prototype_index() (in module protosleepnet.proto_reconstruction.utils) (in module protosleepnet.seed_recon_all_m) build_residual_downstream_fn() (in module protosleepnet.baselines.learn_residual_prototypes_vq) build_sequence_encoder_fn() (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) build_train_loader() (in module protosleepnet.proto_reconstruction.utils) BUILDERS (in module protosleepnet.baselines.test_occlusion_mixer)

C

CH_COLORS (in module protosleepnet.figure_reconstruction.prototype_card) channel_mixer (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) ChannelDropoutWrapper (class in protosleepnet.baselines.channel_dropout) ChannelMixerWrapper (class in protosleepnet.baselines.channel_mixer) ChannelOcclusionWrapper (class in protosleepnet.ablation.test_occlusion) (class in protosleepnet.baselines.test_occlusion) (class in protosleepnet.test_occlusion) CHANNELS (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.demo.evaluate_explanations) (in module protosleepnet.demo.precompute) (in module protosleepnet.extract_embeddings) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.test_pretrained) channels_acc_list (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) channels_to_occlude (protosleepnet.ablation.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.baselines.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.test_occlusion.ChannelOcclusionWrapper attribute) CLASS_NAMES (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_residual_prototypes) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.convert_and_predict_seeds) (in module protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.test_occlusion) (in module protosleepnet.test_prototypes)

classifier (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) (protosleepnet.baselines.residual_model.ResidualSequenceWrapper attribute) clean_score() (in module protosleepnet.demo.precompute_indomain) COLORS (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) compute_band_sensitivity() (in module protosleepnet.figure_reconstruction.rule_learning) compute_class_means() (in module protosleepnet.proto_reconstruction.reconstruction_relevance) compute_event_duration() (in module protosleepnet.figure_reconstruction.prototype_card) compute_event_durations() (in module protosleepnet.figure_reconstruction.rule_learning) compute_l2_sq_distances_np() (in module protosleepnet.proto_reconstruction.utils) compute_metrics() (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_residual_prototypes) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.convert_and_predict_seeds) (in module protosleepnet.plot.vq) (in module protosleepnet.test_occlusion) (in module protosleepnet.test_prototypes) compute_metrics_from_predictions() (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) compute_model() (in module protosleepnet.demo.precompute) compute_signals() (in module protosleepnet.demo.precompute) compute_spectral_similarity() (in module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics) CONFIGS (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.proto_reconstruction.utils) convergence_sweep() (in module protosleepnet.proto_reconstruction.reconstruction_relevance)

D

D_CLF (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer) D_MODEL (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer) d_model (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute) D_SEQ (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch_mixer)

DATASET (in module protosleepnet.demo.precompute) DATASET_CONFIGS (in module protosleepnet.clinical_probing.staging.extract_embeddings) DATASET_NAME_MAP (in module protosleepnet.clinical_probing.staging.extract_embeddings) db_to_lin() (in module protosleepnet.figure_reconstruction.prototype_card) DEFAULT_MODEL (in module protosleepnet.test_pretrained) DEFAULT_REPO (in module protosleepnet.test_pretrained) dominant_stage() (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_stability_reconstruction) DROPOUT (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer)

E

EMB_DIR (in module protosleepnet.proto_reconstruction.utils) embed_subject() (in module protosleepnet.demo.precompute) EMG_BIN_END (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) EMG_BIN_START (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) encode_subject_vq() (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) EOG_BANDS (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) EOG_EXPECTED (in module protosleepnet.figure_reconstruction.rule_learning) epoch_classifier (protosleepnet.baselines.residual_model.ResidualSequenceWrapper attribute) epoch_encoder (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) (protosleepnet.baselines.residual_model.ResidualSequenceWrapper attribute) epoch_ig_subject() (in module protosleepnet.demo.precompute) EPOCH_KWARGS (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_3ch_mixer) eval_epoch() (in module protosleepnet.demo.evaluate_explanations) evaluate_all_subjects() (in module protosleepnet.test_prototype_randomization) evaluate_batch() (in module protosleepnet.ablation.test_occlusion) evaluate_on_dataset() (in module protosleepnet.test_pretrained)

evaluate_subject() (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_residual_prototypes) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.convert_and_predict_seeds) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.test_occlusion) (in module protosleepnet.test_prototypes) evaluate_with_codebook() (in module protosleepnet.seed_stability_pipeline) EXPECTED_TOP_BANDS (in module protosleepnet.demo.evaluate_explanations) (in module protosleepnet.figure_reconstruction.rule_learning) extract_embeddings() (in module protosleepnet.seed_stability_pipeline) extract_epoch_embeddings() (in module protosleepnet.proto_reconstruction.data_driven) extract_epoch_encoder() (in module protosleepnet.baselines.extract_residual_embeddings) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) extract_split() (in module protosleepnet.baselines.extract_residual_embeddings) (in module protosleepnet.extract_proto_embeddings) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings)

F

FACTORY (in module protosleepnet.test_occlusion) FEAT_NAMES (in module protosleepnet.clinical_probing.parkinsons.analyze_features) FEATURES (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) FIG_DIR (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) filterbank (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute) format_s1() (in module protosleepnet.figure_reconstruction.rule_learning) format_s2() (in module protosleepnet.figure_reconstruction.rule_learning) format_s3() (in module protosleepnet.figure_reconstruction.rule_learning)

format_s4() (in module protosleepnet.figure_reconstruction.rule_learning) forward() (protosleepnet.ablation.test_occlusion.ChannelOcclusionWrapper method) (protosleepnet.baselines.channel_dropout.ChannelDropoutWrapper method) (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper method) (protosleepnet.baselines.channel_mixer.SeqEpochEncoder method) (protosleepnet.baselines.channel_mixer.ZeroEmbeddingDropout method) (protosleepnet.baselines.residual_model.ResidualSequenceWrapper method) (protosleepnet.baselines.test_occlusion.ChannelOcclusionWrapper method) (protosleepnet.baselines.test_residual_prototypes.VQResidualModel method) (protosleepnet.posthoc_prototypes.test_posthoc_prototypes.VQWrappedModel method) (protosleepnet.posthoc_prototypes.test_vq_paired.VQWrappedModel method) (protosleepnet.test_occlusion.ChannelOcclusionWrapper method) freq_hz() (in module protosleepnet.figure_reconstruction.prototype_card) FREQ_RES (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning)

G

gaussian_mmd() (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_stability_reconstruction) generate_card() (in module protosleepnet.figure_reconstruction.prototype_card) generate_compact_rule() (in module protosleepnet.figure_reconstruction.rule_learning) generate_per_task_csv() (in module protosleepnet.clinical_probing.parkinsons.generate_tables) generate_rule_text() (in module protosleepnet.figure_reconstruction.prototype_card) generate_summary() (in module protosleepnet.clinical_probing.parkinsons.generate_tables)

get_device() (in module protosleepnet.proto_reconstruction.utils) get_label() (in module protosleepnet.clinical_probing.parkinsons.analyze_features) get_metrics() (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper method) (protosleepnet.baselines.residual_model.ResidualSequenceWrapper method) get_paths() (in module protosleepnet.proto_reconstruction.utils) greek() (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning)

H

HF_NAME (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) HF_REPO_ID (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch_mixer) (in module protosleepnet.train)

hungarian_codebook_alignment() (in module protosleepnet.seed_stability_summary)

I

IGNORE (in module protosleepnet.demo.precompute) (in module protosleepnet.demo.select_subjects)

J

JSON_DIR (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq)

L

LINES (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) load_and_compute() (in module protosleepnet.clinical_probing.parkinsons.analyze_features) load_codebook() (in module protosleepnet.proto_reconstruction.utils) load_data_driven_init() (in module protosleepnet.proto_reconstruction.hybrid) load_frozen_model() (in module protosleepnet.proto_reconstruction.utils) load_local_model() (in module protosleepnet.clinical_probing.staging.extract_embeddings) load_metrics_json() (in module protosleepnet.plot.vq) load_model() (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.extract_proto_embeddings) (in module protosleepnet.learn_prototypes_vq) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.test_prototypes) load_predictions() (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.vq)

load_recon() (in module protosleepnet.seed_mmd_analysis) load_residual_model() (in module protosleepnet.baselines.extract_residual_embeddings) (in module protosleepnet.baselines.learn_residual_prototypes_vq) (in module protosleepnet.baselines.test_residual_prototypes) load_results() (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) load_seed_reconstructions() (in module protosleepnet.seed_stability_reconstruction) load_seed_results() (in module protosleepnet.seed_stability_summary) load_subjects() (in module protosleepnet.probing.alzheimers.diagnosis_probe) (in module protosleepnet.probing.parkinsons.diagnosis_probe) load_subjects_from_dir() (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) load_summary() (in module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics) load_train_embeddings() (in module protosleepnet.proto_reconstruction.utils) load_training_embeddings() (in module protosleepnet.seed_recon_all_m) (in module protosleepnet.seed_stability_pipeline) (in module protosleepnet.test_prototype_randomization) lstm (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute)

M

M_VALUES (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_recon_all_m) main() (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.extract_residual_embeddings) (in module protosleepnet.baselines.learn_residual_prototypes_vq) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_occlusion_mixer) (in module protosleepnet.baselines.test_residual_prototypes) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch_mixer) (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.analyze_features) (in module protosleepnet.clinical_probing.staging.extract_embeddings) (in module protosleepnet.convert_and_predict_seeds) (in module protosleepnet.demo.evaluate_explanations) (in module protosleepnet.demo.precompute) (in module protosleepnet.demo.precompute_indomain) (in module protosleepnet.demo.select_subjects) (in module protosleepnet.demo.subset_bundle) (in module protosleepnet.extract_embeddings) (in module protosleepnet.extract_proto_embeddings) (in module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics) (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) (in module protosleepnet.learn_prototypes_vq) (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) (in module protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans) (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.probing.alzheimers.diagnosis_probe) (in module protosleepnet.probing.parkinsons.diagnosis_probe) (in module protosleepnet.proto_reconstruction.data_driven) (in module protosleepnet.proto_reconstruction.hybrid) (in module protosleepnet.proto_reconstruction.model_driven) (in module protosleepnet.proto_reconstruction.reconstruction_relevance) (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_recon_all_m) (in module protosleepnet.seed_stability_pipeline) (in module protosleepnet.seed_stability_reconstruction) (in module protosleepnet.seed_stability_summary) (in module protosleepnet.test_occlusion) (in module protosleepnet.test_pretrained) (in module protosleepnet.test_prototype_randomization) (in module protosleepnet.test_prototypes) (in module protosleepnet.train) make_prototype_distance_fn() (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.proto_reconstruction.reconstruction_relevance) MARKERS (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) MASK_U8 (in module protosleepnet.demo.precompute) (in module protosleepnet.demo.subset_bundle) mcy (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) MIN_TRAIN_ACC (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) MIXER_KWARGS (in module protosleepnet.test_occlusion) (in module protosleepnet.train) MixerTrainer (class in protosleepnet.baselines.channel_mixer) modality_emb (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) mode (protosleepnet.ablation.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.baselines.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.test_occlusion.ChannelOcclusionWrapper attribute) model (protosleepnet.ablation.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.baselines.channel_dropout.ChannelDropoutWrapper attribute) (protosleepnet.baselines.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.baselines.test_residual_prototypes.VQResidualModel attribute) (protosleepnet.posthoc_prototypes.test_posthoc_prototypes.VQWrappedModel attribute) (protosleepnet.posthoc_prototypes.test_vq_paired.VQWrappedModel attribute) (protosleepnet.test_occlusion.ChannelOcclusionWrapper attribute) MODEL_HF (in module protosleepnet.demo.evaluate_explanations) (in module protosleepnet.demo.select_subjects) (in module protosleepnet.demo.subset_bundle) MODEL_KWARGS (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) MODEL_NAME (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch_mixer) (in module protosleepnet.train)

MODELS (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) (in module protosleepnet.demo.precompute) (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) MODELS_DIR (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.proto_reconstruction.utils) module protosleepnet protosleepnet.ablation protosleepnet.ablation.test_occlusion protosleepnet.ablation.train_ablation protosleepnet.ablation.train_ablation_seqsleepnet protosleepnet.baselines protosleepnet.baselines.channel_dropout protosleepnet.baselines.channel_mixer protosleepnet.baselines.extract_residual_embeddings protosleepnet.baselines.learn_residual_prototypes_vq protosleepnet.baselines.residual_model protosleepnet.baselines.test_occlusion protosleepnet.baselines.test_occlusion_mixer protosleepnet.baselines.test_residual_prototypes protosleepnet.baselines.test_seq_1ch protosleepnet.baselines.test_st_1ch protosleepnet.baselines.train_seq_1ch_residual protosleepnet.baselines.train_seq_3ch protosleepnet.baselines.train_seq_3ch_dropout protosleepnet.baselines.train_seq_3ch_mixer protosleepnet.baselines.train_st_1ch_residual protosleepnet.baselines.train_st_3ch protosleepnet.baselines.train_st_3ch_dropout protosleepnet.baselines.train_st_3ch_mixer protosleepnet.build_protosleepnet protosleepnet.clinical_probing protosleepnet.clinical_probing.alzheimers protosleepnet.clinical_probing.alzheimers.generate_plots protosleepnet.clinical_probing.alzheimers.generate_tables protosleepnet.clinical_probing.parkinsons protosleepnet.clinical_probing.parkinsons.analyze_features protosleepnet.clinical_probing.parkinsons.generate_plots protosleepnet.clinical_probing.parkinsons.generate_tables protosleepnet.clinical_probing.staging protosleepnet.clinical_probing.staging.extract_embeddings protosleepnet.convert_and_predict_seeds protosleepnet.demo protosleepnet.demo.build_cards protosleepnet.demo.evaluate_explanations protosleepnet.demo.pack protosleepnet.demo.precompute protosleepnet.demo.precompute_indomain protosleepnet.demo.select_subjects protosleepnet.demo.subset_bundle protosleepnet.extract_embeddings protosleepnet.extract_proto_embeddings protosleepnet.figure_reconstruction protosleepnet.figure_reconstruction.compute_cross_dataset_metrics protosleepnet.figure_reconstruction.compute_local_explanations protosleepnet.figure_reconstruction.prototype_card protosleepnet.figure_reconstruction.rule_learning protosleepnet.learn_prototypes_vq protosleepnet.plot protosleepnet.plot.occlusion protosleepnet.plot.occlusion_ablation protosleepnet.plot.residual protosleepnet.plot.residual_ablation protosleepnet.plot.vq protosleepnet.posthoc_prototypes protosleepnet.posthoc_prototypes.extract_epoch_embeddings protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans protosleepnet.posthoc_prototypes.probe_posthoc_prototypes protosleepnet.posthoc_prototypes.test_posthoc_prototypes protosleepnet.posthoc_prototypes.test_vq_paired protosleepnet.probing protosleepnet.probing.alzheimers protosleepnet.probing.alzheimers.diagnosis_probe protosleepnet.probing.parkinsons protosleepnet.probing.parkinsons.diagnosis_probe protosleepnet.proto_reconstruction protosleepnet.proto_reconstruction.data_driven protosleepnet.proto_reconstruction.hybrid protosleepnet.proto_reconstruction.model_driven protosleepnet.proto_reconstruction.reconstruction_relevance protosleepnet.proto_reconstruction.utils protosleepnet.seed_mmd_analysis protosleepnet.seed_recon_all_m protosleepnet.seed_stability_pipeline protosleepnet.seed_stability_reconstruction protosleepnet.seed_stability_summary protosleepnet.test_occlusion protosleepnet.test_pretrained protosleepnet.test_prototype_randomization protosleepnet.test_prototypes protosleepnet.train MS (in module protosleepnet.clinical_probing.alzheimers.generate_plots) (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_tables)

N

n_channels (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) N_CLASSES (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer)

n_classes (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) N_FREQ (in module protosleepnet.figure_reconstruction.rule_learning)

O

optimize_all_prototypes() (in module protosleepnet.proto_reconstruction.model_driven)

OUTPUT_BASE (in module protosleepnet.proto_reconstruction.utils)

P

p (protosleepnet.ablation.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.baselines.channel_dropout.ChannelDropoutWrapper attribute) (protosleepnet.baselines.test_occlusion.ChannelOcclusionWrapper attribute) (protosleepnet.test_occlusion.ChannelOcclusionWrapper attribute) p_apply (protosleepnet.baselines.channel_mixer.ZeroEmbeddingDropout attribute) PHYSIOEX_ROOT (in module protosleepnet.figure_reconstruction.prototype_card) (in module protosleepnet.figure_reconstruction.rule_learning) PHYSIOLOGY_CHECKLIST (in module protosleepnet.figure_reconstruction.rule_learning) PIPELINE (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.clinical_probing.staging.extract_embeddings) (in module protosleepnet.extract_embeddings) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.test_pretrained) plot_acc_vs_M() (in module protosleepnet.clinical_probing.parkinsons.generate_plots) plot_backbone() (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) (in module protosleepnet.plot.vq) plot_summary_bar() (in module protosleepnet.clinical_probing.parkinsons.generate_plots) PRED_DIR (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) predict_classes() (in module protosleepnet.proto_reconstruction.reconstruction_relevance) predict_subject() (in module protosleepnet.demo.precompute) PREFERRED_EOG_BAND (in module protosleepnet.figure_reconstruction.rule_learning) print_summary_table() (in module protosleepnet.test_pretrained) PROTO_FILES (in module protosleepnet.demo.subset_bundle) protosleepnet module protosleepnet.ablation module protosleepnet.ablation.test_occlusion module protosleepnet.ablation.train_ablation module protosleepnet.ablation.train_ablation_seqsleepnet module protosleepnet.baselines module protosleepnet.baselines.channel_dropout module protosleepnet.baselines.channel_mixer module protosleepnet.baselines.extract_residual_embeddings module protosleepnet.baselines.learn_residual_prototypes_vq module protosleepnet.baselines.residual_model module protosleepnet.baselines.test_occlusion module protosleepnet.baselines.test_occlusion_mixer module protosleepnet.baselines.test_residual_prototypes module protosleepnet.baselines.test_seq_1ch module protosleepnet.baselines.test_st_1ch module protosleepnet.baselines.train_seq_1ch_residual module protosleepnet.baselines.train_seq_3ch module protosleepnet.baselines.train_seq_3ch_dropout module protosleepnet.baselines.train_seq_3ch_mixer module protosleepnet.baselines.train_st_1ch_residual module protosleepnet.baselines.train_st_3ch module protosleepnet.baselines.train_st_3ch_dropout module protosleepnet.baselines.train_st_3ch_mixer module protosleepnet.build_protosleepnet module protosleepnet.clinical_probing module protosleepnet.clinical_probing.alzheimers module protosleepnet.clinical_probing.alzheimers.generate_plots module protosleepnet.clinical_probing.alzheimers.generate_tables module protosleepnet.clinical_probing.parkinsons module protosleepnet.clinical_probing.parkinsons.analyze_features module protosleepnet.clinical_probing.parkinsons.generate_plots module protosleepnet.clinical_probing.parkinsons.generate_tables module protosleepnet.clinical_probing.staging module protosleepnet.clinical_probing.staging.extract_embeddings module

protosleepnet.convert_and_predict_seeds module protosleepnet.demo module protosleepnet.demo.build_cards module protosleepnet.demo.evaluate_explanations module protosleepnet.demo.pack module protosleepnet.demo.precompute module protosleepnet.demo.precompute_indomain module protosleepnet.demo.select_subjects module protosleepnet.demo.subset_bundle module protosleepnet.extract_embeddings module protosleepnet.extract_proto_embeddings module protosleepnet.figure_reconstruction module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics module protosleepnet.figure_reconstruction.compute_local_explanations module protosleepnet.figure_reconstruction.prototype_card module protosleepnet.figure_reconstruction.rule_learning module protosleepnet.learn_prototypes_vq module protosleepnet.plot module protosleepnet.plot.occlusion module protosleepnet.plot.occlusion_ablation module protosleepnet.plot.residual module protosleepnet.plot.residual_ablation module protosleepnet.plot.vq module protosleepnet.posthoc_prototypes module protosleepnet.posthoc_prototypes.extract_epoch_embeddings module protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes module protosleepnet.posthoc_prototypes.test_posthoc_prototypes module protosleepnet.posthoc_prototypes.test_vq_paired module protosleepnet.probing module protosleepnet.probing.alzheimers module protosleepnet.probing.alzheimers.diagnosis_probe module protosleepnet.probing.parkinsons module protosleepnet.probing.parkinsons.diagnosis_probe module protosleepnet.proto_reconstruction module protosleepnet.proto_reconstruction.data_driven module protosleepnet.proto_reconstruction.hybrid module protosleepnet.proto_reconstruction.model_driven module protosleepnet.proto_reconstruction.reconstruction_relevance module protosleepnet.proto_reconstruction.utils module protosleepnet.seed_mmd_analysis module protosleepnet.seed_recon_all_m module protosleepnet.seed_stability_pipeline module protosleepnet.seed_stability_reconstruction module protosleepnet.seed_stability_summary module protosleepnet.test_occlusion module protosleepnet.test_pretrained module protosleepnet.test_prototype_randomization module protosleepnet.test_prototypes module protosleepnet.train module

Q

quantize_proba_u8() (in module protosleepnet.demo.pack)

R

RECON (in module protosleepnet.figure_reconstruction.compute_local_explanations) RECON_FILES (in module protosleepnet.demo.precompute_indomain) ResidualSequenceWrapper (class in protosleepnet.baselines.residual_model) ResidualTrainer (class in protosleepnet.baselines.residual_model)

ResidualTrainerSingleLoss (class in protosleepnet.baselines.residual_model) resolve_output_dir() (in module protosleepnet.proto_reconstruction.utils) run_loso_fold() (in module protosleepnet.probing.alzheimers.diagnosis_probe) (in module protosleepnet.probing.parkinsons.diagnosis_probe)

S

save_prototype_results() (in module protosleepnet.proto_reconstruction.utils) save_summary() (in module protosleepnet.proto_reconstruction.utils) SCENARIO_LABELS (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) SCENARIOS (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.test_occlusion) (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) (in module protosleepnet.test_occlusion) score_subjects() (in module protosleepnet.demo.select_subjects) select() (in module protosleepnet.demo.select_subjects) SENSITIVITY_SOFT_FLOOR (in module protosleepnet.figure_reconstruction.rule_learning) SEQ_KWARGS (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_3ch_mixer) SEQ_LEN (in module protosleepnet.ablation.test_occlusion) (in module protosleepnet.baselines.learn_residual_prototypes_vq) (in module protosleepnet.baselines.test_seq_1ch) (in module protosleepnet.baselines.test_st_1ch) (in module protosleepnet.extract_embeddings) (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired)

SeqEpochEncoder (class in protosleepnet.baselines.channel_mixer) sequence_encoder (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) (protosleepnet.baselines.residual_model.ResidualSequenceWrapper attribute) SLEEP_CLASS_NAMES (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) STAGE_COLORS (in module protosleepnet.clinical_probing.parkinsons.analyze_features) (in module protosleepnet.figure_reconstruction.prototype_card) STAGE_NAMES (in module protosleepnet.demo.build_cards) (in module protosleepnet.demo.precompute) (in module protosleepnet.demo.select_subjects) (in module protosleepnet.demo.subset_bundle) (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.figure_reconstruction.rule_learning) (in module protosleepnet.proto_reconstruction.reconstruction_relevance) (in module protosleepnet.proto_reconstruction.utils) (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_recon_all_m) (in module protosleepnet.seed_stability_reconstruction) STAGES (in module protosleepnet.clinical_probing.parkinsons.analyze_features) subject_accuracy() (in module protosleepnet.posthoc_prototypes.test_vq_paired)

T

TABLES_DIR (in module protosleepnet.clinical_probing.alzheimers.generate_tables) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) TASKS (in module protosleepnet.clinical_probing.parkinsons.analyze_features) (in module protosleepnet.clinical_probing.parkinsons.generate_plots) (in module protosleepnet.clinical_probing.parkinsons.generate_tables) test_baseline() (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) test_nmf() (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) test_vq() (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes)

TRAIN_CONFIG (in module protosleepnet.baselines.train_seq_1ch_residual) (in module protosleepnet.baselines.train_seq_3ch) (in module protosleepnet.baselines.train_seq_3ch_dropout) (in module protosleepnet.baselines.train_seq_3ch_mixer) (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch) (in module protosleepnet.baselines.train_st_3ch_dropout) (in module protosleepnet.baselines.train_st_3ch_mixer) trim_window() (in module protosleepnet.demo.precompute)

U

update_channel_acc() (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper method)

V

VQ_METHOD (in module protosleepnet.plot.residual) (in module protosleepnet.plot.residual_ablation) VQ_METHODS (in module protosleepnet.plot.vq)

VQResidualModel (class in protosleepnet.baselines.test_residual_prototypes) VQWrappedModel (class in protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (class in protosleepnet.posthoc_prototypes.test_vq_paired)

W

write_epoch_ig() (in module protosleepnet.demo.precompute) write_f16() (in module protosleepnet.demo.pack) write_f32() (in module protosleepnet.demo.pack)

write_i16() (in module protosleepnet.demo.pack) write_reconstructions() (in module protosleepnet.demo.precompute) write_u16() (in module protosleepnet.demo.pack) write_u8() (in module protosleepnet.demo.pack)

Z

zero_emb_dropout (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) zero_init_gru() (in module protosleepnet.baselines.residual_model)

zero_init_transformer() (in module protosleepnet.baselines.residual_model) ZeroEmbeddingDropout (class in protosleepnet.baselines.channel_mixer)