Index

A

activation (koopman_graph.encoder.BaseGNNModule attribute) activation_name (koopman_graph.encoder.BaseGNNModule attribute) adjacency_to_edge_index() (in module koopman_graph.datasets.metr_la) adjacency_to_edge_weight() (in module koopman_graph.datasets.metr_la) aggregate_mae (koopman_graph.metrics.EvaluationResult attribute)

aggregate_mape (koopman_graph.metrics.EvaluationResult attribute) aggregate_rmse (koopman_graph.metrics.EvaluationResult attribute) allow_dynamic_topology (koopman_graph.data.GraphSnapshotSequence property) AnisotropicAdvectionGridBenchmark (class in koopman_graph.datasets) (class in koopman_graph.datasets.grid)

B

backward (koopman_graph.training.LossWeights attribute) (koopman_graph.training.TrainingLossBreakdown attribute) backward_loss (koopman_graph.training.FitHistory attribute) BackwardConsistencyLoss (class in koopman_graph.losses) BaseGNNModule (class in koopman_graph.encoder)

best_epoch (koopman_graph.training.FitHistory attribute) best_loss (koopman_graph.training.FitHistory attribute) build_adjacency_matrix() (in module koopman_graph.datasets.metr_la) build_checkpoint() (in module koopman_graph.serialization) build_model_config() (in module koopman_graph.serialization) build_traffic_cache_payload() (in module koopman_graph.datasets.metr_la)

C

compute_backward_consistency_sequence_loss() (in module koopman_graph.training) compute_eigenvalue_regularization_loss() (in module koopman_graph.training) compute_forward_consistency_sequence_loss() (in module koopman_graph.training) compute_rollout_loss() (in module koopman_graph.training) compute_sequence_loss() (in module koopman_graph.training) compute_spectrum() (in module koopman_graph.analysis) compute_training_loss() (in module koopman_graph.training)

constant_loss_weights() (in module koopman_graph.training) control_at() (koopman_graph.data.GraphSnapshotSequence method) control_dim (koopman_graph.data.GraphSnapshotSequence property) (koopman_graph.operator.KoopmanOperator attribute) control_inputs (koopman_graph.data.GraphSnapshotSequence property) control_term() (koopman_graph.operator.KoopmanOperator method) convs (koopman_graph.encoder.BaseGNNModule attribute)

D

decode_mode_shapes() (in module koopman_graph.analysis) decoder (koopman_graph.model.GraphKoopmanModel attribute) dense_inverse_matrix() (koopman_graph.operator.KoopmanOperator method) DMDBaseline (class in koopman_graph.baselines)

DMDcBaseline (class in koopman_graph.baselines) download_distances_csv() (in module koopman_graph.datasets.metr_la) download_matpower_case118() (in module koopman_graph.datasets.ieee118) download_sensor_ids() (in module koopman_graph.datasets.metr_la) dropout (koopman_graph.encoder.GATEncoder attribute)

E

edge_index (koopman_graph.data.GraphSnapshotSequence property) edge_weight (koopman_graph.data.GraphSnapshotSequence property) EDMDBaseline (class in koopman_graph.baselines) eigenvalue (koopman_graph.training.LossWeights attribute) (koopman_graph.training.TrainingLossBreakdown attribute) eigenvalue_loss (koopman_graph.training.FitHistory attribute) EigenvalueRegularizationLoss (class in koopman_graph.losses) eigenvalues (koopman_graph.analysis.KoopmanSpectrum attribute)

eigenvectors (koopman_graph.analysis.KoopmanSpectrum attribute) encoder (koopman_graph.model.GraphKoopmanModel attribute) ensure_topology_cache() (in module koopman_graph.datasets.ieee118) ensure_traffic_cache() (in module koopman_graph.datasets.metr_la) epochs (koopman_graph.training.FitHistory attribute) eval_one_epoch() (in module koopman_graph.training) evaluate() (koopman_graph.model.GraphKoopmanModel method) evaluate_forecast() (in module koopman_graph.metrics) EvaluationResult (class in koopman_graph.metrics)

F

fit() (koopman_graph.baselines.DMDBaseline method) (koopman_graph.baselines.DMDcBaseline method) (koopman_graph.baselines.EDMDBaseline method) (koopman_graph.model.GraphKoopmanModel method) FitHistory (class in koopman_graph.training) forward (koopman_graph.training.LossWeights attribute) (koopman_graph.training.TrainingLossBreakdown attribute) forward() (koopman_graph.encoder.BaseGNNModule method) (koopman_graph.losses.BackwardConsistencyLoss method) (koopman_graph.losses.EigenvalueRegularizationLoss method) (koopman_graph.losses.ForwardConsistencyLoss method) (koopman_graph.model.GraphKoopmanModel method) (koopman_graph.operator.KoopmanOperator method)

forward_loss (koopman_graph.training.FitHistory attribute) ForwardConsistencyLoss (class in koopman_graph.losses) frequencies (koopman_graph.analysis.KoopmanSpectrum attribute) from_arrays() (koopman_graph.data.GraphSnapshotSequence class method) from_dynamic_arrays() (koopman_graph.data.GraphSnapshotSequence class method)

G

GATEncoder (class in koopman_graph.encoder) generate() (koopman_graph.datasets.AnisotropicAdvectionGridBenchmark class method) (koopman_graph.datasets.grid.AnisotropicAdvectionGridBenchmark class method) (koopman_graph.datasets.grid.GridDynamicGraphBenchmark class method) (koopman_graph.datasets.GridDynamicGraphBenchmark class method) (koopman_graph.datasets.ieee118.IEEE118DynamicBenchmark class method) (koopman_graph.datasets.IEEE118DynamicBenchmark class method) (koopman_graph.datasets.synthetic.SyntheticDynamicGraphBenchmark class method) (koopman_graph.datasets.SyntheticDynamicGraphBenchmark class method)

GNNDecoder (class in koopman_graph.decoder) GNNEncoder (class in koopman_graph.encoder) GraphKoopmanModel (class in koopman_graph.model) GraphSnapshotSequence (class in koopman_graph.data) grid_node_index() (in module koopman_graph.datasets.grid) GridDynamicGraphBenchmark (class in koopman_graph.datasets) (class in koopman_graph.datasets.grid) growth_rates (koopman_graph.analysis.KoopmanSpectrum attribute)

H

has_controls (koopman_graph.data.GraphSnapshotSequence property) heads (koopman_graph.encoder.GATEncoder attribute) hidden_channels (koopman_graph.decoder.GNNDecoder attribute) (koopman_graph.encoder.GATEncoder attribute) (koopman_graph.encoder.GNNEncoder attribute)

horizon (koopman_graph.metrics.HorizonMetrics attribute) HorizonMetrics (class in koopman_graph.metrics) horizons (koopman_graph.metrics.EvaluationResult attribute)

I

IEEE118DynamicBenchmark (class in koopman_graph.datasets) (class in koopman_graph.datasets.ieee118) in_channels (koopman_graph.data.GraphSnapshotSequence property) IN_CHANNELS (koopman_graph.datasets.ieee118.IEEE118DynamicBenchmark attribute) (koopman_graph.datasets.IEEE118DynamicBenchmark attribute) (koopman_graph.datasets.metr_la.MetrLaTrafficBenchmark attribute) (koopman_graph.datasets.MetrLaTrafficBenchmark attribute) in_channels (koopman_graph.encoder.GATEncoder attribute) (koopman_graph.encoder.GNNEncoder attribute)

init_mode (koopman_graph.operator.KoopmanOperator attribute) init_scale (koopman_graph.operator.KoopmanOperator attribute) input_channels (koopman_graph.encoder.BaseGNNModule attribute) input_dim_name (koopman_graph.encoder.BaseGNNModule attribute) inverse_step() (koopman_graph.operator.KoopmanOperator method) is_dynamic_topology (koopman_graph.data.GraphSnapshotSequence property) is_sequence_of_sequences() (in module koopman_graph.training) iter_epoch() (koopman_graph.data.WindowSampler method)

K

K (koopman_graph.operator.KoopmanOperator property) koopman (koopman_graph.model.GraphKoopmanModel attribute) koopman_graph.analysis module koopman_graph.baselines module koopman_graph.data module koopman_graph.datasets module koopman_graph.datasets.grid module koopman_graph.datasets.ieee118 module koopman_graph.datasets.metr_la module koopman_graph.datasets.synthetic module

koopman_graph.decoder module koopman_graph.encoder module koopman_graph.losses module koopman_graph.metrics module koopman_graph.model module koopman_graph.operator module koopman_graph.serialization module koopman_graph.training module KoopmanOperator (class in koopman_graph.operator) KoopmanSpectrum (class in koopman_graph.analysis)

L

latent_dim (koopman_graph.decoder.GNNDecoder attribute) (koopman_graph.encoder.GATEncoder attribute) (koopman_graph.encoder.GNNEncoder attribute) (koopman_graph.model.GraphKoopmanModel attribute) (koopman_graph.operator.KoopmanOperator attribute) linear_ramp_loss_weights() (in module koopman_graph.training) load() (koopman_graph.model.GraphKoopmanModel class method) load_checkpoint() (in module koopman_graph.serialization) load_sequence() (koopman_graph.datasets.metr_la.MetrLaTrafficBenchmark class method) (koopman_graph.datasets.MetrLaTrafficBenchmark class method)

load_topology() (in module koopman_graph.datasets.ieee118) (koopman_graph.datasets.ieee118.IEEE118DynamicBenchmark class method) (koopman_graph.datasets.IEEE118DynamicBenchmark class method) (koopman_graph.datasets.metr_la.MetrLaTrafficBenchmark class method) (koopman_graph.datasets.MetrLaTrafficBenchmark class method) load_traffic_cache() (in module koopman_graph.datasets.metr_la) loss (koopman_graph.training.FitHistory attribute) LossWeights (class in koopman_graph.training)

M

mae (koopman_graph.metrics.HorizonMetrics attribute) mae() (in module koopman_graph.metrics) magnitudes (koopman_graph.analysis.KoopmanSpectrum attribute) mape (koopman_graph.metrics.HorizonMetrics attribute) mape() (in module koopman_graph.metrics) max_spectral_radius (koopman_graph.operator.KoopmanOperator attribute) mean_training_loss_breakdown() (in module koopman_graph.training) MetrLaTrafficBenchmark (class in koopman_graph.datasets) (class in koopman_graph.datasets.metr_la) mode_amplitudes() (koopman_graph.analysis.KoopmanSpectrum method) module koopman_graph.analysis koopman_graph.baselines koopman_graph.data koopman_graph.datasets koopman_graph.datasets.grid koopman_graph.datasets.ieee118 koopman_graph.datasets.metr_la koopman_graph.datasets.synthetic koopman_graph.decoder koopman_graph.encoder koopman_graph.losses koopman_graph.metrics koopman_graph.model koopman_graph.operator koopman_graph.serialization koopman_graph.training

N

normalize_speeds() (in module koopman_graph.datasets.metr_la) NUM_BUSES (koopman_graph.datasets.ieee118.IEEE118DynamicBenchmark attribute) (koopman_graph.datasets.IEEE118DynamicBenchmark attribute) num_layers (koopman_graph.encoder.BaseGNNModule attribute) num_nodes (koopman_graph.data.GraphSnapshotSequence property)

num_origins (koopman_graph.metrics.EvaluationResult attribute) NUM_SENSORS (koopman_graph.datasets.metr_la.MetrLaTrafficBenchmark attribute) (koopman_graph.datasets.MetrLaTrafficBenchmark attribute) num_timesteps (koopman_graph.data.GraphSnapshotSequence property) num_windows (koopman_graph.data.WindowSampler property)

O

one_step_loss() (in module koopman_graph.training)

out_channels (koopman_graph.decoder.GNNDecoder attribute)

P

parameterization (koopman_graph.operator.KoopmanOperator attribute) parse_matpower_case() (in module koopman_graph.datasets.ieee118) predict() (koopman_graph.baselines.DMDBaseline method) (koopman_graph.baselines.DMDcBaseline method) (koopman_graph.baselines.EDMDBaseline method) (koopman_graph.model.GraphKoopmanModel method)

preprocess_speeds() (in module koopman_graph.datasets.metr_la)

R

read_h5_speed_window() (in module koopman_graph.datasets.metr_la) reconstruct_model() (in module koopman_graph.serialization) reconstruction (koopman_graph.training.LossWeights attribute) (koopman_graph.training.TrainingLossBreakdown attribute) reconstruction_loss (koopman_graph.training.FitHistory attribute) reset_control_parameters() (koopman_graph.operator.KoopmanOperator method) reset_parameters() (koopman_graph.operator.KoopmanOperator method) resolve_device() (in module koopman_graph.training) resolve_early_stopping_monitor() (in module koopman_graph.training) resolve_loss_weights_for_epoch() (in module koopman_graph.training) resolve_lr_scheduler() (in module koopman_graph.training)

resolve_rollout_start_indices() (in module koopman_graph.training) resolve_sequence() (in module koopman_graph.data) resolve_training_sequences() (in module koopman_graph.training) resolve_validation_sequences() (in module koopman_graph.training) rmse (koopman_graph.metrics.HorizonMetrics attribute) rmse() (in module koopman_graph.metrics) rollout (koopman_graph.training.LossWeights attribute) (koopman_graph.training.TrainingLossBreakdown attribute) rollout_controls() (koopman_graph.data.GraphSnapshotSequence method) rollout_loss (koopman_graph.training.FitHistory attribute) rollout_multi_start_loss() (in module koopman_graph.losses) rollout_sequence_loss() (in module koopman_graph.losses)

S

save() (koopman_graph.model.GraphKoopmanModel method) save_checkpoint() (in module koopman_graph.serialization) should_stop_early() (in module koopman_graph.training) slice() (koopman_graph.data.GraphSnapshotSequence method) snapshot_state_dict() (in module koopman_graph.serialization) snapshots (koopman_graph.data.GraphSnapshotSequence property) spectral_radius() (koopman_graph.operator.KoopmanOperator method)

spectrum() (koopman_graph.baselines.DMDBaseline method) (koopman_graph.baselines.DMDcBaseline method) (koopman_graph.baselines.EDMDBaseline method) (koopman_graph.model.GraphKoopmanModel method) stopped_early (koopman_graph.training.FitHistory attribute) SyntheticDynamicGraphBenchmark (class in koopman_graph.datasets) (class in koopman_graph.datasets.synthetic)

T

temporal_split() (in module koopman_graph.data) TemporalSplit (class in koopman_graph.data) test (koopman_graph.data.TemporalSplit attribute) time_step (koopman_graph.analysis.KoopmanSpectrum attribute) (koopman_graph.model.GraphKoopmanModel attribute) to_floats() (koopman_graph.training.TrainingLossBreakdown method)

topology_from_matpower_text() (in module koopman_graph.datasets.ieee118) total (koopman_graph.training.TrainingLossBreakdown attribute) train (koopman_graph.data.TemporalSplit attribute) train_one_epoch() (in module koopman_graph.training) train_windowed_epoch() (in module koopman_graph.training) TrainingLossBreakdown (class in koopman_graph.training)

V

val (koopman_graph.data.TemporalSplit attribute) val_backward_loss (koopman_graph.training.FitHistory attribute) val_eigenvalue_loss (koopman_graph.training.FitHistory attribute)

val_forward_loss (koopman_graph.training.FitHistory attribute) val_loss (koopman_graph.training.FitHistory attribute) val_reconstruction_loss (koopman_graph.training.FitHistory attribute) val_rollout_loss (koopman_graph.training.FitHistory attribute)

W

WindowSampler (class in koopman_graph.data)

Z

zeros() (koopman_graph.training.TrainingLossBreakdown class method)