Index

A

aibedo module aibedo.models module

aibedo.models.MLP module aibedo.models.unet module AIBEDO_MLP (class in aibedo.models.MLP)

B

BaseModel (class in aibedo.models)

F

forward() (aibedo.models.BaseModel method) (aibedo.models.MLP.AIBEDO_MLP method) (aibedo.models.unet.SphericalUNet method)

G

get_datamodule() (in module aibedo.interface)

get_model() (in module aibedo.interface) get_model_and_data() (in module aibedo.interface)

M

module aibedo aibedo.models aibedo.models.MLP aibedo.models.unet

P

postprocess_raw_predictions() (aibedo.models.BaseModel method)

predict() (aibedo.models.BaseModel method)

R

raw_outputs_to_denormalized_per_variable_dict() (aibedo.models.BaseModel method) raw_predict() (aibedo.models.BaseModel method)

reload_and_test_model() (in module aibedo.test) reload_model_from_config_and_ckpt() (in module aibedo.interface) run_model() (in module aibedo.train)

S

SphericalUNet (class in aibedo.models.unet)