Index

A

add_common_args() (in module protosleepnet.proto_reconstruction.utils) ALL_M (in module protosleepnet.clinical_probing.atlas) analysis_A() (in module protosleepnet.clinical_probing.structure) analysis_B() (in module protosleepnet.clinical_probing.structure) analyze_one_m() (in module protosleepnet.seed_mmd_analysis) anon_label() (in module protosleepnet.demo.precompute_indomain) anon_labels() (in module protosleepnet.demo.subset_bundle) ARRAYS (in module protosleepnet.demo.subset_bundle) as_variant() (in module protosleepnet.clinical_probing.nested_loso)

assign() (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.posthoc_prototypes.codebook_refinement) attention (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute) attn_pool (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) auc() (in module protosleepnet.figure_reconstruction.detect_events) auc_ci() (in module protosleepnet.clinical_probing.inference) audit_model() (in module protosleepnet.demo.evaluate_explanations) available_M() (in module protosleepnet.clinical_probing.atlas)

B

B_REM (in module protosleepnet.clinical_probing.structure) BACKBONE (in module protosleepnet.figure_reconstruction.detect_events) BACKBONE_CONFIGS (in module protosleepnet.train) BACKBONE_MODEL (in module protosleepnet.posthoc_prototypes.codebook_refinement) BAND_GREEK (in module protosleepnet.figure_reconstruction.rule_learning) BAND_NAMES (in module protosleepnet.demo.evaluate_explanations) band_power() (in module protosleepnet.figure_reconstruction.detect_events) BANDS (in module protosleepnet.demo.evaluate_explanations) (in module protosleepnet.figure_reconstruction.events_relevance) BASE (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) bh() (in module protosleepnet.clinical_probing.nested_loso) (in module protosleepnet.clinical_probing.structure) bh_adjust() (in module protosleepnet.clinical_probing.inference) BLOCK (in module protosleepnet.clinical_probing.atlas) bonferroni() (in module protosleepnet.clinical_probing.inference)

build() (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.subject_stats) 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) build_downstream_fn() (in module protosleepnet.learn_prototypes_vq) build_full_loader() (in module protosleepnet.proto_reconstruction.utils) build_grid() (in module protosleepnet.clinical_probing.nested_loso) 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)

C

CAL_QUANTILES (in module protosleepnet.clinical_probing.coverage_features) CHANNEL (in module protosleepnet.figure_reconstruction.detect_events) 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) check_alignment() (in module protosleepnet.figure_reconstruction.detect_events) CHIN_CHANNELS (in module protosleepnet.clinical_probing.subject_stats) 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) COHORT_PAIRS (in module protosleepnet.clinical_probing.structure) COHORTS (in module protosleepnet.clinical_probing.inference) (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) 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) comparator_path() (in module protosleepnet.clinical_probing.hypnogram_comparator) compare() (in module protosleepnet.clinical_probing.inference) (in module protosleepnet.posthoc_prototypes.quantization_disagreement) compute_band_sensitivity() (in module protosleepnet.figure_reconstruction.rule_learning) compute_event_durations() (in module protosleepnet.figure_reconstruction.rule_learning) compute_features() (in module protosleepnet.clinical_probing.build_features) 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) CONFIRMATORY (in module protosleepnet.clinical_probing.inference) confirmatory() (in module protosleepnet.clinical_probing.inference) criteria() (in module protosleepnet.figure_reconstruction.threshold_sweep) CRITERION (in module protosleepnet.clinical_probing.subject_stats)

D

d (protosleepnet.clinical_probing.nested_loso.FixedVariant attribute) (protosleepnet.clinical_probing.nested_loso.UnionVariant attribute) 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) DEFAULT_MODEL (in module protosleepnet.test_pretrained) default_out() (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.clinical_probing.inference) (in module protosleepnet.clinical_probing.nested_loso) (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.clinical_probing.subject_stats)

DEFAULT_RECON (in module protosleepnet.figure_reconstruction.threshold_sweep) DEFAULT_RECONSTRUCTIONS (in module protosleepnet.figure_reconstruction.events_relevance) default_reconstructions() (in module protosleepnet.clinical_probing.structure) DEFAULT_REPO (in module protosleepnet.test_pretrained) DEFAULT_ROOT (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.clinical_probing.hypnogram_comparator) (in module protosleepnet.clinical_probing.inference) (in module protosleepnet.clinical_probing.nested_loso) (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.clinical_probing.subject_stats) delong() (in module protosleepnet.clinical_probing.inference) delta_auc_boot() (in module protosleepnet.clinical_probing.nested_loso) derangement() (in module protosleepnet.figure_reconstruction.stage_permutation) detect() (in module protosleepnet.figure_reconstruction.detect_events) diff_ci() (in module protosleepnet.clinical_probing.structure) differences() (in module protosleepnet.posthoc_prototypes.non_inferiority) dominant() (in module protosleepnet.posthoc_prototypes.codebook_refinement) dominant_stage() (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.seed_mmd_analysis) draw_figure() (in module protosleepnet.figure_reconstruction.threshold_sweep) DROPOUT (in module protosleepnet.baselines.train_st_1ch_residual) (in module protosleepnet.baselines.train_st_3ch_mixer)

E

effective_weights() (in module protosleepnet.clinical_probing.nested_loso) EMB_DIR (in module protosleepnet.proto_reconstruction.utils) embed_subject() (in module protosleepnet.demo.precompute) EMG_BIN_END (in module protosleepnet.figure_reconstruction.rule_learning) EMG_BIN_START (in module protosleepnet.figure_reconstruction.rule_learning) encode_subject_vq() (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) entropy() (in module protosleepnet.clinical_probing.coverage_features) EOG_BANDS (in module protosleepnet.figure_reconstruction.rule_learning) EOG_EXPECTED (in module protosleepnet.figure_reconstruction.rule_learning) EPOCH (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) 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) EPOCH_MIN (in module protosleepnet.posthoc_prototypes.quantization_disagreement) EPS (in module protosleepnet.clinical_probing.coverage_features) estimators() (in module protosleepnet.clinical_probing.nested_loso) 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_cohort() (in module protosleepnet.clinical_probing.inference) 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_COUNTS (in module protosleepnet.figure_reconstruction.threshold_sweep) 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.posthoc_prototypes.extract_epoch_embeddings)

F

FACTORY (in module protosleepnet.test_occlusion) feature_matrix() (in module protosleepnet.clinical_probing.coverage_features) FEATURE_NAMES (in module protosleepnet.clinical_probing.coverage_features) feature_path() (in module protosleepnet.clinical_probing.build_features) features_34() (in module protosleepnet.clinical_probing.hypnogram_comparator) FIG_DIR (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) fit_transform() (protosleepnet.clinical_probing.nested_loso.FixedVariant method) (protosleepnet.clinical_probing.nested_loso.NamedFixed method) (protosleepnet.clinical_probing.nested_loso.UnionVariant method) FixedVariant (class in protosleepnet.clinical_probing.nested_loso) 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_RES (in module protosleepnet.figure_reconstruction.rule_learning) FS (in module protosleepnet.figure_reconstruction.detect_events)

G

gaussian_mmd() (in module protosleepnet.seed_mmd_analysis) generate_compact_rule() (in module protosleepnet.figure_reconstruction.rule_learning) get_device() (in module protosleepnet.proto_reconstruction.utils) 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.rule_learning) GRID (in module protosleepnet.clinical_probing.nested_loso) (in module protosleepnet.figure_reconstruction.threshold_sweep)

H

HALF_WIDTH (in module protosleepnet.figure_reconstruction.threshold_sweep) hardware_variant() (in module protosleepnet.clinical_probing.subject_stats) hedges_g() (in module protosleepnet.clinical_probing.structure) 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)

holm() (in module protosleepnet.clinical_probing.nested_loso) HOUR (in module protosleepnet.clinical_probing.hypnogram_comparator) hungarian_codebook_alignment() (in module protosleepnet.seed_stability_summary) HW_VARIANT (in module protosleepnet.clinical_probing.subject_stats)

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)

K

kappa() (in module protosleepnet.posthoc_prototypes.non_inferiority)

L

labels_path() (in module protosleepnet.clinical_probing.build_features) latex() (in module protosleepnet.clinical_probing.inference) LIGHTS (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) lights_off_epoch() (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) load() (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.posthoc_prototypes.non_inferiority) (in module protosleepnet.posthoc_prototypes.quantization_disagreement) load_atlas() (in module protosleepnet.clinical_probing.atlas) load_codebook() (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.posthoc_prototypes.codebook_refinement) (in module protosleepnet.proto_reconstruction.utils) load_cohort() (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) 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_metadata() (in module protosleepnet.clinical_probing.subject_stats) 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.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_records() (in module protosleepnet.figure_reconstruction.threshold_sweep) load_relevance() (in module protosleepnet.figure_reconstruction.events_relevance) 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_seed_results() (in module protosleepnet.seed_stability_summary) load_signal() (in module protosleepnet.figure_reconstruction.detect_events) load_stats() (in module protosleepnet.clinical_probing.build_features) load_subjects_from_dir() (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) load_summary() (in module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics) load_trace() (in module protosleepnet.clinical_probing.inference) load_train_embeddings() (in module protosleepnet.proto_reconstruction.utils) load_training() (in module protosleepnet.clinical_probing.atlas) load_training_embeddings() (in module protosleepnet.seed_recon_all_m) (in module protosleepnet.seed_stability_pipeline) (in module protosleepnet.test_prototype_randomization) log() (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.clinical_probing.nested_loso) loso_matrices() (in module protosleepnet.clinical_probing.nested_loso) lstm (protosleepnet.baselines.channel_mixer.SeqEpochEncoder attribute)

M

M_DEMO (in module protosleepnet.clinical_probing.structure) M_MAX (in module protosleepnet.clinical_probing.nested_loso) M_ROBUST (in module protosleepnet.clinical_probing.structure) 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) macro_f1_ci() (in module protosleepnet.clinical_probing.inference) 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_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.atlas) (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.clinical_probing.coverage_features) (in module protosleepnet.clinical_probing.hypnogram_comparator) (in module protosleepnet.clinical_probing.inference) (in module protosleepnet.clinical_probing.nested_loso) (in module protosleepnet.clinical_probing.staging.extract_embeddings) (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.clinical_probing.subject_stats) (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.figure_reconstruction.compute_cross_dataset_metrics) (in module protosleepnet.figure_reconstruction.compute_local_explanations) (in module protosleepnet.figure_reconstruction.detect_events) (in module protosleepnet.figure_reconstruction.events_relevance) (in module protosleepnet.figure_reconstruction.rule_learning) (in module protosleepnet.figure_reconstruction.stage_permutation) (in module protosleepnet.figure_reconstruction.threshold_sweep) (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.codebook_refinement) (in module protosleepnet.posthoc_prototypes.extract_epoch_embeddings) (in module protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans) (in module protosleepnet.posthoc_prototypes.non_inferiority) (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.quantization_disagreement) (in module protosleepnet.posthoc_prototypes.test_posthoc_prototypes) (in module protosleepnet.posthoc_prototypes.test_vq_paired) (in module protosleepnet.proto_reconstruction.data_driven) (in module protosleepnet.proto_reconstruction.hybrid) (in module protosleepnet.proto_reconstruction.model_driven) (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_recon_all_m) (in module protosleepnet.seed_stability_pipeline) (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_clf() (in module protosleepnet.clinical_probing.nested_loso) make_prototype_distance_fn() (in module protosleepnet.figure_reconstruction.compute_local_explanations) margin() (in module protosleepnet.figure_reconstruction.threshold_sweep) 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) material() (in module protosleepnet.clinical_probing.structure) matrix() (in module protosleepnet.clinical_probing.hypnogram_comparator) mcy (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper attribute) METRICS (in module protosleepnet.figure_reconstruction.events_relevance) metrics() (in module protosleepnet.clinical_probing.nested_loso) MIN_PAGES (in module protosleepnet.figure_reconstruction.events_relevance) 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 (in module protosleepnet.clinical_probing.atlas) (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.figure_reconstruction.events_relevance) 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.demo.precompute) (in module protosleepnet.figure_reconstruction.threshold_sweep) (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.non_inferiority) (in module protosleepnet.posthoc_prototypes.quantization_disagreement) 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_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.atlas protosleepnet.clinical_probing.build_features protosleepnet.clinical_probing.coverage_features protosleepnet.clinical_probing.hypnogram_comparator protosleepnet.clinical_probing.inference protosleepnet.clinical_probing.nested_loso protosleepnet.clinical_probing.parkinsons protosleepnet.clinical_probing.staging protosleepnet.clinical_probing.staging.extract_embeddings protosleepnet.clinical_probing.structure protosleepnet.clinical_probing.subject_stats 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.figure_reconstruction protosleepnet.figure_reconstruction.compute_cross_dataset_metrics protosleepnet.figure_reconstruction.compute_local_explanations protosleepnet.figure_reconstruction.detect_events protosleepnet.figure_reconstruction.events_relevance protosleepnet.figure_reconstruction.rule_learning protosleepnet.figure_reconstruction.stage_permutation protosleepnet.figure_reconstruction.threshold_sweep 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.codebook_refinement protosleepnet.posthoc_prototypes.extract_epoch_embeddings protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans protosleepnet.posthoc_prototypes.non_inferiority protosleepnet.posthoc_prototypes.probe_posthoc_prototypes protosleepnet.posthoc_prototypes.quantization_disagreement protosleepnet.posthoc_prototypes.test_posthoc_prototypes protosleepnet.posthoc_prototypes.test_vq_paired protosleepnet.proto_reconstruction protosleepnet.proto_reconstruction.data_driven protosleepnet.proto_reconstruction.hybrid protosleepnet.proto_reconstruction.model_driven protosleepnet.proto_reconstruction.utils protosleepnet.seed_mmd_analysis protosleepnet.seed_recon_all_m protosleepnet.seed_stability_pipeline protosleepnet.seed_stability_summary protosleepnet.test_occlusion protosleepnet.test_pretrained protosleepnet.test_prototype_randomization protosleepnet.test_prototypes protosleepnet.train MS (in module protosleepnet.posthoc_prototypes.codebook_refinement) (in module protosleepnet.posthoc_prototypes.non_inferiority)

N

N_BOOT (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.posthoc_prototypes.non_inferiority) N_BOOT_F1 (in module protosleepnet.clinical_probing.inference) 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_EDGES (in module protosleepnet.clinical_probing.structure)

N_FREQ (in module protosleepnet.figure_reconstruction.rule_learning) N_REP (in module protosleepnet.clinical_probing.coverage_features) N_SLICES (in module protosleepnet.clinical_probing.coverage_features) N_SUBSAMPLE (in module protosleepnet.clinical_probing.atlas) NAME (in module protosleepnet.clinical_probing.build_features) (in module protosleepnet.clinical_probing.nested_loso) NamedFixed (class in protosleepnet.clinical_probing.nested_loso) NAMES (in module protosleepnet.clinical_probing.hypnogram_comparator) names (protosleepnet.clinical_probing.nested_loso.NamedFixed attribute) (protosleepnet.clinical_probing.nested_loso.UnionVariant attribute) nested() (in module protosleepnet.clinical_probing.inference)

O

one_at_a_time() (in module protosleepnet.figure_reconstruction.threshold_sweep) optimize_all_prototypes() (in module protosleepnet.proto_reconstruction.model_driven) ORDER (in module protosleepnet.figure_reconstruction.events_relevance)

ORIGINAL_NAMES (in module protosleepnet.clinical_probing.coverage_features) outcomes() (in module protosleepnet.figure_reconstruction.threshold_sweep) 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) PAD (in module protosleepnet.figure_reconstruction.detect_events) PAGE (in module protosleepnet.figure_reconstruction.detect_events) pair() (in module protosleepnet.posthoc_prototypes.non_inferiority) pair_excess() (in module protosleepnet.clinical_probing.structure) pair_stats() (in module protosleepnet.posthoc_prototypes.codebook_refinement) paired() (in module protosleepnet.clinical_probing.nested_loso) PCA_NARROW (in module protosleepnet.clinical_probing.nested_loso) PCA_WIDE (in module protosleepnet.clinical_probing.nested_loso) PER_MIN (in module protosleepnet.figure_reconstruction.events_relevance) per_page() (in module protosleepnet.figure_reconstruction.detect_events) 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_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) PRED_DIR (in module protosleepnet.plot.occlusion) (in module protosleepnet.plot.occlusion_ablation) predict_subject() (in module protosleepnet.demo.precompute) PREFERRED_EOG_BAND (in module protosleepnet.figure_reconstruction.rule_learning) print_summary() (in module protosleepnet.posthoc_prototypes.quantization_disagreement) 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_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.atlas module protosleepnet.clinical_probing.build_features module protosleepnet.clinical_probing.coverage_features module protosleepnet.clinical_probing.hypnogram_comparator module protosleepnet.clinical_probing.inference module protosleepnet.clinical_probing.nested_loso module protosleepnet.clinical_probing.parkinsons module

protosleepnet.clinical_probing.staging module protosleepnet.clinical_probing.staging.extract_embeddings module protosleepnet.clinical_probing.structure module protosleepnet.clinical_probing.subject_stats 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.figure_reconstruction module protosleepnet.figure_reconstruction.compute_cross_dataset_metrics module protosleepnet.figure_reconstruction.compute_local_explanations module protosleepnet.figure_reconstruction.detect_events module protosleepnet.figure_reconstruction.events_relevance module protosleepnet.figure_reconstruction.rule_learning module protosleepnet.figure_reconstruction.stage_permutation module protosleepnet.figure_reconstruction.threshold_sweep 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.codebook_refinement module protosleepnet.posthoc_prototypes.extract_epoch_embeddings module protosleepnet.posthoc_prototypes.learn_prototypes_vq_kmeans module protosleepnet.posthoc_prototypes.non_inferiority module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes module protosleepnet.posthoc_prototypes.quantization_disagreement module protosleepnet.posthoc_prototypes.test_posthoc_prototypes module protosleepnet.posthoc_prototypes.test_vq_paired 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.utils module protosleepnet.seed_mmd_analysis module protosleepnet.seed_recon_all_m module protosleepnet.seed_stability_pipeline 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 PUBLISHED (in module protosleepnet.figure_reconstruction.threshold_sweep)

Q

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

R

rank() (in module protosleepnet.figure_reconstruction.threshold_sweep) rarefaction_budget() (in module protosleepnet.clinical_probing.coverage_features) RECON (in module protosleepnet.figure_reconstruction.compute_local_explanations) RECON_FILES (in module protosleepnet.demo.precompute_indomain) REM (in module protosleepnet.clinical_probing.structure) rem_share_tests() (in module protosleepnet.clinical_probing.structure) ResidualSequenceWrapper (class in protosleepnet.baselines.residual_model)

ResidualTrainer (class in protosleepnet.baselines.residual_model) ResidualTrainerSingleLoss (class in protosleepnet.baselines.residual_model) resolve_codebook() (in module protosleepnet.figure_reconstruction.detect_events) resolve_output_dir() (in module protosleepnet.proto_reconstruction.utils) resolve_psg_dir() (in module protosleepnet.figure_reconstruction.detect_events) ROWS (in module protosleepnet.clinical_probing.inference) run() (in module protosleepnet.clinical_probing.structure) run_classification() (in module protosleepnet.clinical_probing.nested_loso)

S

save_prototype_results() (in module protosleepnet.proto_reconstruction.utils) save_summary() (in module protosleepnet.proto_reconstruction.utils) save_trace() (in module protosleepnet.clinical_probing.nested_loso) 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) SEED (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.figure_reconstruction.events_relevance) (in module protosleepnet.posthoc_prototypes.non_inferiority) segments() (in module protosleepnet.clinical_probing.coverage_features) select() (in module protosleepnet.demo.select_subjects) selected_variant_shares() (in module protosleepnet.clinical_probing.nested_loso) SENSITIVITY_SOFT_FLOOR (in module protosleepnet.figure_reconstruction.rule_learning) SENSITIVITY_SUBJECT (in module protosleepnet.figure_reconstruction.events_relevance) 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) SHIFTS (in module protosleepnet.figure_reconstruction.detect_events) SHORT (in module protosleepnet.clinical_probing.structure) SLEEP_CLASS_NAMES (in module protosleepnet.posthoc_prototypes.probe_posthoc_prototypes) slice_directions() (in module protosleepnet.clinical_probing.coverage_features) sliced_w1() (in module protosleepnet.clinical_probing.coverage_features) spearman() (in module protosleepnet.figure_reconstruction.events_relevance) spectrum() (in module protosleepnet.clinical_probing.structure) STAGE (in module protosleepnet.posthoc_prototypes.codebook_refinement) 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.utils) (in module protosleepnet.seed_mmd_analysis) (in module protosleepnet.seed_recon_all_m) STAGES (in module protosleepnet.clinical_probing.structure) (in module protosleepnet.figure_reconstruction.threshold_sweep) (in module protosleepnet.posthoc_prototypes.quantization_disagreement) STAT_KEYS (in module protosleepnet.clinical_probing.subject_stats) statistics() (in module protosleepnet.figure_reconstruction.stage_permutation) STATS_M (in module protosleepnet.clinical_probing.build_features) stats_path() (in module protosleepnet.clinical_probing.subject_stats) Subject (class in protosleepnet.clinical_probing.build_features) subject_accuracy() (in module protosleepnet.posthoc_prototypes.test_vq_paired) subject_features() (in module protosleepnet.clinical_probing.coverage_features)

T

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) thresholds_at_tolerance() (in module protosleepnet.figure_reconstruction.threshold_sweep) TOLERANCES (in module protosleepnet.figure_reconstruction.threshold_sweep) 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)

training_dirs() (in module protosleepnet.clinical_probing.atlas) transitions() (in module protosleepnet.clinical_probing.coverage_features) trim_window() (in module protosleepnet.demo.precompute)

U

UnionVariant (class in protosleepnet.clinical_probing.nested_loso)

update_channel_acc() (protosleepnet.baselines.channel_mixer.ChannelMixerWrapper method) USES (in module protosleepnet.figure_reconstruction.threshold_sweep)

V

variants (protosleepnet.clinical_probing.nested_loso.UnionVariant attribute) verdict() (in module protosleepnet.figure_reconstruction.threshold_sweep) verdict_matrix() (in module protosleepnet.figure_reconstruction.stage_permutation) 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

wilson_ci() (in module protosleepnet.clinical_probing.inference) WINDOW (in module protosleepnet.figure_reconstruction.detect_events) (in module protosleepnet.posthoc_prototypes.codebook_refinement) window_mask() (in module protosleepnet.clinical_probing.subject_stats) (in module protosleepnet.posthoc_prototypes.codebook_refinement) 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_labels() (in module protosleepnet.clinical_probing.build_features) write_reconstructions() (in module protosleepnet.demo.precompute) write_u16() (in module protosleepnet.demo.pack) write_u8() (in module protosleepnet.demo.pack)

X

X (protosleepnet.clinical_probing.nested_loso.FixedVariant attribute)

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)