Index

_

__init__() (rtnn.dataset.DataPreprocessor method) (rtnn.evaluater.MetricTracker method) (rtnn.evaluater.NMAELoss method) (rtnn.evaluater.NMSELoss method) (rtnn.logger.Logger method) (rtnn.model_utils.ModelUtils method) (rtnn.models.fcn.FCBlock method) (rtnn.models.fcn.FCN method) (rtnn.models.mlp.MLP method) (rtnn.models.mlp.MLPBlock method) (rtnn.models.mlp.MLPResidual method) (rtnn.models.pinn.LayerPositionalEmbedding method) (rtnn.models.pinn.PINN method) (rtnn.models.rnn.BaseRNN method) (rtnn.models.rnn.RNN_GRU method) (rtnn.models.rnn.RNN_LSTM method) (rtnn.models.transformer.EncoderTorch method) (rtnn.utils.FileUtils method)

B

BaseRNN (class in rtnn.models.rnn)

bn (rtnn.models.fcn.FCBlock attribute)

C

calc_abs() (in module rtnn.evaluater) clear_logs() (rtnn.logger.Logger method) conservation_residual() (in module rtnn.evaluater) cosz (rtnn.dataset.DataPreprocessor attribute)

count (rtnn.evaluater.MetricTracker attribute) count_parameters_by_layer() (rtnn.model_utils.ModelUtils static method) create_datasets_and_loaders() (in module rtnn.main) create_normalization_mapping() (in module rtnn.main)

D

DataPreprocessor (class in rtnn.dataset) dfs (rtnn.dataset.DataPreprocessor attribute)

dim_change (rtnn.models.fcn.FCN attribute) dim_expand (rtnn.models.fcn.FCN attribute)

E

EasyDict (class in rtnn.utils) EncoderTorch (class in rtnn.models.transformer)

error() (rtnn.logger.Logger method) etime (rtnn.dataset.DataPreprocessor attribute) exception() (rtnn.logger.Logger method)

F

FCBlock (class in rtnn.models.fcn) FCN (class in rtnn.models.fcn) feature_channel (rtnn.models.fcn.FCN attribute) FileUtils (class in rtnn.utils) final (rtnn.models.rnn.BaseRNN attribute) forward() (rtnn.evaluater.NMAELoss method) (rtnn.evaluater.NMSELoss method) (rtnn.models.fcn.FCBlock method) (rtnn.models.fcn.FCN method) (rtnn.models.mlp.MLP method) (rtnn.models.mlp.MLPBlock method) (rtnn.models.mlp.MLPResidual method) (rtnn.models.pinn.LayerPositionalEmbedding method) (rtnn.models.pinn.PINN method) (rtnn.models.rnn.BaseRNN method) (rtnn.models.transformer.EncoderTorch method)

G

get_data_files() (in module rtnn.main) get_loss_function() (in module rtnn.evaluater) get_parameter_number() (rtnn.model_utils.ModelUtils static method) get_version() (in module rtnn.version)

getmean() (rtnn.evaluater.MetricTracker method) getsqrtmean() (rtnn.evaluater.MetricTracker method) getstd() (rtnn.evaluater.MetricTracker method) gmrae_all() (in module rtnn.evaluater)

H

heating_rate() (in module rtnn.evaluater)

hidden_layers (rtnn.models.fcn.FCN attribute) hidden_size (rtnn.models.rnn.BaseRNN attribute)

I

info() (rtnn.logger.Logger method) init_hidden() (rtnn.models.rnn.BaseRNN method)

initialize_model() (in module rtnn.main) input_layer (rtnn.models.fcn.FCN attribute)

L

lai (rtnn.dataset.DataPreprocessor attribute) LayerPositionalEmbedding (class in rtnn.models.pinn) linear (rtnn.models.fcn.FCBlock attribute) load_checkpoint() (rtnn.model_utils.ModelUtils static method) load_checkpoint_if_requested() (in module rtnn.main) load_model() (in module rtnn.model_loader)

load_training_checkpoint() (rtnn.model_utils.ModelUtils static method) log_configuration() (in module rtnn.main) log_metrics() (rtnn.logger.Logger method) log_model_summary() (rtnn.model_utils.ModelUtils static method) Logger (class in rtnn.logger) logger (rtnn.dataset.DataPreprocessor attribute)

M

mae_all() (in module rtnn.evaluater) main() (in module rtnn.main) makedir() (rtnn.utils.FileUtils static method) makefile() (rtnn.utils.FileUtils static method) mare_all() (in module rtnn.evaluater) mbe_all() (in module rtnn.evaluater) MetricTracker (class in rtnn.evaluater) min_dims (rtnn.dataset.DataPreprocessor attribute) MLP (class in rtnn.models.mlp) MLPBlock (class in rtnn.models.mlp) MLPResidual (class in rtnn.models.mlp) ModelUtils (class in rtnn.model_utils) module rtnn.dataset rtnn.diagnostics rtnn.evaluater rtnn.logger rtnn.main rtnn.model_loader rtnn.model_utils rtnn.models.fcn rtnn.models.mlp rtnn.models.pinn rtnn.models.rnn rtnn.models.transformer rtnn.utils rtnn.version

mse_all() (in module rtnn.evaluater)

N

nmae_all() (in module rtnn.evaluater) NMAELoss (class in rtnn.evaluater) nmse_all() (in module rtnn.evaluater) NMSELoss (class in rtnn.evaluater)

norm_mapping (rtnn.dataset.DataPreprocessor attribute) normalization_type (rtnn.dataset.DataPreprocessor attribute) normalize() (rtnn.dataset.DataPreprocessor method) num_layers (rtnn.models.rnn.BaseRNN attribute)

O

output_channel (rtnn.models.fcn.FCN attribute) (rtnn.models.rnn.BaseRNN attribute)

output_layer (rtnn.models.fcn.FCN attribute) ov (rtnn.dataset.DataPreprocessor attribute)

P

parse_args() (in module rtnn.main) parse_years() (in module rtnn.main) physics_loss() (in module rtnn.evaluater) PINN (class in rtnn.models.pinn) plot_all_diagnostics() (in module rtnn.diagnostics) plot_flux_and_abs() (in module rtnn.diagnostics)

plot_flux_and_abs_lines() (in module rtnn.diagnostics) plot_loss_histories() (in module rtnn.diagnostics) plot_metric_histories() (in module rtnn.diagnostics) plot_spatial_temporal_density() (in module rtnn.diagnostics) print_model_layers() (rtnn.model_utils.ModelUtils static method) print_version() (in module rtnn.main)

R

r2_all() (in module rtnn.evaluater) relu (rtnn.models.fcn.FCBlock attribute) reset() (rtnn.evaluater.MetricTracker method) rnn (rtnn.models.rnn.BaseRNN attribute) RNN_GRU (class in rtnn.models.rnn) RNN_LSTM (class in rtnn.models.rnn) rs (rtnn.dataset.DataPreprocessor attribute) rtnn.dataset module rtnn.diagnostics module rtnn.evaluater module rtnn.logger module rtnn.main module rtnn.model_loader module

rtnn.model_utils module rtnn.models.fcn module rtnn.models.mlp module rtnn.models.pinn module rtnn.models.rnn module rtnn.models.transformer module rtnn.utils module rtnn.version module run_validation() (in module rtnn.evaluater)

S

save_checkpoint() (rtnn.model_utils.ModelUtils static method) save_emergency_checkpoint() (rtnn.model_utils.ModelUtils static method) save_training_checkpoint() (rtnn.model_utils.ModelUtils static method) sbatch (rtnn.dataset.DataPreprocessor attribute) seq_length (rtnn.models.fcn.FCN attribute) setup_device_and_seed() (in module rtnn.main) setup_directories_and_logging() (in module rtnn.main)

show_header() (rtnn.logger.Logger method) ssa (rtnn.dataset.DataPreprocessor attribute) start_task() (rtnn.logger.Logger method) stats() (in module rtnn.diagnostics) step() (rtnn.logger.Logger method) stime (rtnn.dataset.DataPreprocessor attribute) subplots() (in module rtnn.diagnostics) success() (rtnn.logger.Logger method)

T

tbatch (rtnn.dataset.DataPreprocessor attribute) time_blocks (rtnn.dataset.DataPreprocessor attribute)

train_epoch() (in module rtnn.main) tstep (rtnn.dataset.DataPreprocessor attribute)

U

unnorm_mpas() (in module rtnn.evaluater)

update() (rtnn.evaluater.MetricTracker method)

V

value (rtnn.evaluater.MetricTracker attribute)

value_sq (rtnn.evaluater.MetricTracker attribute)

W

warning() (rtnn.logger.Logger method)

Y

years (rtnn.dataset.DataPreprocessor attribute)