Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | R | S | T | V | W _ __init__() (Config method) (CVConfig method) (DatasetConfig method) (DatasetInfo method) (DiagonalNSDConv method), [1] (FilteredWikipediaDataset method) (GeneralNSDConv method), [1] (HardwareConfig method) (ModelConfig method) (NSDModel method), [1] (NSDVariant method) (OptimConfig method) (OrthogonalNSDConv method), [1] (RegConfig method) (Registry method) (SheafDataModule method) (SheafLightningModule method) (SplitsConfig method) (WandBConfig method) __new__() (ModelType method) (ModelVariant method) A allow_zero_length_dataloader_with_multiple_devices (SheafDataModule attribute) alpha (ModelConfig attribute) apply_split() (in module exp.splits) B batch_size (OptimConfig attribute) build_kwargs() (NSDVariant method) C clamp_val (ModelConfig attribute) Config (class in exp.config) configure_optimizers() (SheafLightningModule method) cuda (HardwareConfig attribute) cv (Config attribute) CVConfig (class in exp.config) D dataset (Config attribute) DatasetConfig (class in exp.config) DatasetInfo (class in exp.data) datasets (SplitsConfig attribute) DIAGONAL (ModelVariant attribute) (NSDVariant attribute) DiagonalNSDConv (class in sheaf_mpnn.nsd) download() (FilteredWikipediaDataset method) download_canonical_splits() (in module exp.gen_splits) dropout (RegConfig attribute) E early_stopping (OptimConfig attribute) enabled (WandBConfig attribute) entity (WandBConfig attribute) epochs (OptimConfig attribute) exp.config module exp.data module exp.gen_splits module exp.module module exp.registries.base module exp.registries.presets module exp.run module exp.splits module exp.sweeps.sweep module F FilteredWikipediaDataset (class in exp.data) folds (CVConfig attribute) (SplitsConfig attribute) forward() (NSDModel method) G GENERAL (ModelVariant attribute) (NSDVariant attribute) GENERAL_ATTENTION (ModelVariant attribute) (NSDVariant attribute) GeneralNSDConv (class in sheaf_mpnn.nsd) generate_config() (in module exp.registries.presets) generate_splits() (in module exp.gen_splits) get() (Registry method) get_map_products() (DiagonalNSDConv method) (GeneralNSDConv method) (OrthogonalNSDConv method) get_or_default() (PresetRegistry method) H hardware (Config attribute) HardwareConfig (class in exp.config) hidden_dim (ModelConfig attribute) homophily (SheafDataModule property) I info (SheafDataModule property) input_dropout (RegConfig attribute) J jknet (ModelConfig attribute) L layer_class (NSDVariant property) layer_kwargs (NSDVariant property) leaky_relu_slope (ModelConfig attribute) list_keys() (Registry method) load_dataset() (in module exp.data) LOW_RANK (ModelVariant attribute) (NSDVariant attribute) lr (OptimConfig attribute) M main() (in module exp.gen_splits) (in module exp.run) (in module exp.sweeps.sweep) message() (DiagonalNSDConv method) (OrthogonalNSDConv method) metric (DatasetInfo attribute) min_acc (CVConfig attribute) model (Config attribute) ModelConfig (class in exp.config) ModelType (class in exp.config) ModelVariant (class in exp.config) module exp.config exp.data exp.gen_splits exp.module exp.registries.base exp.registries.presets exp.run exp.splits exp.sweeps.sweep N name (DatasetConfig attribute) (DatasetInfo attribute) normalize_output (ModelConfig attribute) NSD (ModelType attribute) NSDModel (class in sheaf_mpnn.nsd) NSDVariant (class in sheaf_mpnn.nsd) num_classes (DatasetInfo attribute) num_edges (SheafDataModule property) num_features (DatasetInfo attribute) num_heads (ModelConfig attribute) num_layers (ModelConfig attribute) num_nodes (SheafDataModule property) num_splits (DatasetInfo attribute) num_workers (HardwareConfig attribute) O optim (Config attribute) OptimConfig (class in exp.config) orth_strategy (ModelConfig attribute) ORTHOGONAL (ModelVariant attribute) (NSDVariant attribute) ORTHOGONAL_ATTENTION (ModelVariant attribute) (NSDVariant attribute) OrthogonalNSDConv (class in sheaf_mpnn.nsd) overwrite (SplitsConfig attribute) P persistent_workers (HardwareConfig attribute) pin_memory (HardwareConfig attribute) prepare_data_per_node (SheafDataModule attribute) PresetRegistry (class in exp.registries.presets) process() (FilteredWikipediaDataset method) processed_file_names (FilteredWikipediaDataset property) project (WandBConfig attribute) R rank (ModelConfig attribute) raw_file_names (FilteredWikipediaDataset property) reg (Config attribute) RegConfig (class in exp.config) register() (Registry method) Registry (class in exp.registries.base) reset_parameters() (NSDModel method) root (DatasetConfig attribute) (SplitsConfig attribute) run() (in module exp.run) S seed (CVConfig attribute) setup() (SheafDataModule method) SheafDataModule (class in exp.data) SheafLightningModule (class in exp.module) source (SplitsConfig attribute) split_sizes (SheafDataModule property) split_type (DatasetInfo attribute) splits() (in module exp.gen_splits) splits_dir (SplitsConfig attribute) SplitsConfig (class in exp.gen_splits) stalk_dim (ModelConfig attribute) stop_strategy (OptimConfig attribute) sweep() (in module exp.sweeps.sweep) T test_dataloader() (SheafDataModule method) test_step() (SheafLightningModule method) train_dataloader() (SheafDataModule method) training_step() (SheafLightningModule method) type (ModelConfig attribute) V val_dataloader() (SheafDataModule method) validation_step() (SheafLightningModule method) variant (ModelConfig attribute) W wandb (Config attribute) WandBConfig (class in exp.config) weight_decay (OptimConfig attribute)