Index

_

__call__() (src.FRAME_FM.transforms.BaseTransform method) (src.FRAME_FM.transforms.FillMissingValueTransform method) (src.FRAME_FM.transforms.NormalizeTransform method) (src.FRAME_FM.transforms.RenameTransform method) (src.FRAME_FM.transforms.ResampleTransform method) (src.FRAME_FM.transforms.ReshapeTransform method) (src.FRAME_FM.transforms.ReverseAxisTransform method) (src.FRAME_FM.transforms.RollTransform method) (src.FRAME_FM.transforms.SortAxisTransform method) (src.FRAME_FM.transforms.SqueezeTransform method) (src.FRAME_FM.transforms.SubsetTransform method) (src.FRAME_FM.transforms.TilerTransform method) (src.FRAME_FM.transforms.ToDataArray method) (src.FRAME_FM.transforms.ToTensorTransform method) (src.FRAME_FM.transforms.transforms.BaseTransform method) (src.FRAME_FM.transforms.transforms.FillMissingValueTransform method) (src.FRAME_FM.transforms.transforms.NormalizeTransform method) (src.FRAME_FM.transforms.transforms.RenameTransform method) (src.FRAME_FM.transforms.transforms.ResampleTransform method) (src.FRAME_FM.transforms.transforms.ReshapeTransform method) (src.FRAME_FM.transforms.transforms.ReverseAxisTransform method) (src.FRAME_FM.transforms.transforms.RollTransform method) (src.FRAME_FM.transforms.transforms.SortAxisTransform method) (src.FRAME_FM.transforms.transforms.SqueezeTransform method) (src.FRAME_FM.transforms.transforms.SubsetTransform method) (src.FRAME_FM.transforms.transforms.TilerTransform method) (src.FRAME_FM.transforms.transforms.ToDataArray method) (src.FRAME_FM.transforms.transforms.ToTensorTransform method) (src.FRAME_FM.transforms.transforms.TransposeTransform method) (src.FRAME_FM.transforms.transforms.VarsToDimensionTransform method) (src.FRAME_FM.transforms.TransposeTransform method) (src.FRAME_FM.transforms.VarsToDimensionTransform method)

__getitem__() (src.FRAME_FM.datasets.base_dataset.BaseDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGeoTIFFDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedTimeSeriesDataset method) (src.FRAME_FM.datasets.base_shapefile_dataset.BaseShapefileDataset method) (src.FRAME_FM.datasets.ImageLabel_Dataset.TransformedDataset method) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset method) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputDataset method) __len__() (src.FRAME_FM.datasets.base_dataset.BaseDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGeoTIFFDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedDataset method) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedTimeSeriesDataset method) (src.FRAME_FM.datasets.base_shapefile_dataset.BaseShapefileDataset method) (src.FRAME_FM.datasets.ImageLabel_Dataset.TransformedDataset method) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset method) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputDataset method)

A

apply_preprocessors (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms) apply_transforms() (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms)

ascending (src.FRAME_FM.transforms.SortAxisTransform attribute) (src.FRAME_FM.transforms.transforms.SortAxisTransform attribute)

B

base (src.FRAME_FM.datasets.ImageLabel_Dataset.TransformedDataset attribute) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset attribute) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputDataset attribute) base_ch (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) BaseASCIIGridDataset (class in src.FRAME_FM.datasets.base_gridded_dataset) BaseDataModule (class in src.FRAME_FM.utils.LightningDataModuleWrapper) BaseDataset (class in src.FRAME_FM.datasets.base_dataset) BaseEmbedder (class in src.FRAME_FM.utils.embedders) BaseGeoTIFFDataset (class in src.FRAME_FM.datasets.base_gridded_dataset) BaseGriddedDataset (class in src.FRAME_FM.datasets.base_gridded_dataset) BaseGriddedTimeSeriesDataset (class in src.FRAME_FM.datasets.base_gridded_dataset)

BaseModule (class in src.FRAME_FM.utils.LightningModuleWrapper) BaseShapefileDataset (class in src.FRAME_FM.datasets.base_shapefile_dataset) BaseTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) batch_size (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) blocks (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) bn1 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) bn2 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) bn3 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) boundary (src.FRAME_FM.transforms.TilerTransform attribute) (src.FRAME_FM.transforms.transforms.TilerTransform attribute) BoundedPatchEmbed (class in src.FRAME_FM.utils.embedders)

C

cache_data_to_zarr() (in module src.FRAME_FM.utils.data_utils) cache_dir (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) (src.FRAME_FM.utils.settings.DatasetSettings attribute) calc_embed_omega() (in module src.FRAME_FM.utils.embedders) check_object_type() (in module src.FRAME_FM.utils.transform_utils) CHESSMetGriddedTimeSeriesDataset (class in src.FRAME_FM.datasets.chessmet_dataset) chunks (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedTimeSeriesDataset attribute) (src.FRAME_FM.utils.settings.DatasetSettings attribute) (src.FRAME_FM.utils.settings.DefaultSettings attribute) cls_token (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) CogHandler (class in src.FRAME_FM.dataloaders.cog_dataloader) CombinedDataModule (class in src.FRAME_FM.dataloaders.combined_dataloader)

configure_optimizers() (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method) (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) conv1 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) conv2 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) conv3 (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) ConvAutoencoder (class in src.FRAME_FM.models.demo_convAE) ConvAutoencoderWithLocation (class in src.FRAME_FM.models.demo_convAE) convert_subset_selectors_to_slices() (in module src.FRAME_FM.utils.data_utils) convert_to_long_lat() (in module src.FRAME_FM.dataloaders.demo_geotiff_dataloader) create_cache_path() (in module src.FRAME_FM.utils.data_utils) create_mlflow_logger() (in module src.FRAME_FM.training.logger) (in module src.FRAME_FM.training.logging_utils) create_zarr_name() (in module src.FRAME_FM.utils.data_utils)

D

DA (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms) data (src.FRAME_FM.datasets.base_shapefile_dataset.BaseShapefileDataset attribute) data_root (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) data_uri (in module src.FRAME_FM.datasets.base_shapefile_dataset) (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) (src.FRAME_FM.datasets.base_shapefile_dataset.BaseShapefileDataset attribute) datamodules (src.FRAME_FM.dataloaders.combined_dataloader.CombinedDataModule attribute) dataset (in module src.FRAME_FM.datasets.topsoil_dataset) DatasetSettings (class in src.FRAME_FM.utils.settings) dc_mode_embeddings (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) DEBUG (in module src.FRAME_FM.utils.settings) decoder (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder attribute) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) decoder_blocks (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) decoder_embed (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute)

decoder_norm (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) decoder_omegas (src.FRAME_FM.utils.embedders.STPatchEmbed attribute) DEFAULT_CHUNKS (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedTimeSeriesDataset attribute) default_mask_ratio (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) DefaultSettings (class in src.FRAME_FM.utils.settings) dim (src.FRAME_FM.transforms.ResampleTransform attribute) (src.FRAME_FM.transforms.ReverseAxisTransform attribute) (src.FRAME_FM.transforms.RollTransform attribute) (src.FRAME_FM.transforms.SortAxisTransform attribute) (src.FRAME_FM.transforms.transforms.ResampleTransform attribute) (src.FRAME_FM.transforms.transforms.ReverseAxisTransform attribute) (src.FRAME_FM.transforms.transforms.RollTransform attribute) (src.FRAME_FM.transforms.transforms.SortAxisTransform attribute) dropout (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) DS (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms)

E

ec_mode_embeddings (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) embed_dim (src.FRAME_FM.utils.embedders.BaseEmbedder attribute) (src.FRAME_FM.utils.embedders.PatchEmbed attribute) encoder (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder attribute) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) encoder_omegas (src.FRAME_FM.utils.embedders.STPatchEmbed attribute)

ERA5GriddedTimeSeriesDataset (class in src.FRAME_FM.datasets.era5_dataset) EuroSATAutoencoder (class in src.FRAME_FM.models.demo_autoencoder) EuroSATDataModule (class in src.FRAME_FM.dataloaders.demo_eurosat) exclusion_vars (src.FRAME_FM.transforms.transforms.VarsToDimensionTransform attribute) (src.FRAME_FM.transforms.VarsToDimensionTransform attribute) extract_position_tensor() (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.GeotiffBoundedDataModule method) (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.GeotiffSpatialDataModule method)

F

feature_cols (src.FRAME_FM.dataloaders.gridded_txt_dataloader.GriddedDataModule attribute) (src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule attribute) filename (src.FRAME_FM.dataloaders.gridded_txt_dataloader.GriddedDataModule attribute) (src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule attribute) fill_value (src.FRAME_FM.transforms.FillMissingValueTransform attribute) (src.FRAME_FM.transforms.transforms.FillMissingValueTransform attribute) FillMissingValueTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) FillNaNTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) force_recache (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) forward() (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method) (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse method) (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) (src.FRAME_FM.utils.embedders.BaseEmbedder method) (src.FRAME_FM.utils.embedders.PatchEmbed method) (src.FRAME_FM.utils.embedders.STPatchEmbed method)

forward_decoder() (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) forward_encoder() (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) forward_loss() (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) freq (src.FRAME_FM.transforms.ResampleTransform attribute) (src.FRAME_FM.transforms.transforms.ResampleTransform attribute) from_latent (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder attribute) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute)

G

generate_stats (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) GeotiffBoundedDataModule (class in src.FRAME_FM.dataloaders.demo_geotiff_dataloader) GeotiffSpatialDataModule (class in src.FRAME_FM.dataloaders.demo_geotiff_dataloader) get_main_vars() (in module src.FRAME_FM.utils.data_utils)

get_xr_kwargs() (in module src.FRAME_FM.utils.data_utils) glob_string (src.FRAME_FM.dataloaders.cog_dataloader.CogHandler attribute) GriddedDataModule (class in src.FRAME_FM.dataloaders.gridded_txt_dataloader) (class in src.FRAME_FM.dataloaders.xarray_dataloader)

H

handle_special_uri_case() (in module src.FRAME_FM.utils.data_utils)

hash_preprocessors() (in module src.FRAME_FM.utils.data_utils) head (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute)

I

in_channels (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) initialize_weights() (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) (src.FRAME_FM.utils.embedders.BaseEmbedder method) (src.FRAME_FM.utils.embedders.PatchEmbed method) (src.FRAME_FM.utils.embedders.STPatchEmbed method)

input_embedders (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) input_tile_buffer (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute)

K

k_size (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute)

KERCHUNK_ZIP (in module src.FRAME_FM.models.demo_simple_conv)

L

LandCoverMapGriddedDataset (class in src.FRAME_FM.datasets.land_cover_map_dataset) latent_buffer (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) latent_dim (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) learning_rate (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) list_files() (src.FRAME_FM.dataloaders.cog_dataloader.CogHandler method)

load_data_from_uri() (in module src.FRAME_FM.utils.data_utils) load_files() (src.FRAME_FM.dataloaders.cog_dataloader.CogHandler method) log_metrics() (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) loss_fn (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder attribute) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute)

M

main() (in module src.FRAME_FM.dataloaders.demo_geotiff_dataloader) (in module src.FRAME_FM.dataloaders.gridded_txt_dataloader) (in module src.FRAME_FM.dataloaders.xarray_dataloader) (in module src.FRAME_FM.training.train) (src.FRAME_FM.dataloaders.cog_dataloader.CogHandler method) mask_token (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) max_latents_per_epoch (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) mean (src.FRAME_FM.transforms.NormalizeTransform attribute) (src.FRAME_FM.transforms.transforms.NormalizeTransform attribute) method (src.FRAME_FM.transforms.FillMissingValueTransform attribute) (src.FRAME_FM.transforms.ResampleTransform attribute) (src.FRAME_FM.transforms.transforms.FillMissingValueTransform attribute) (src.FRAME_FM.transforms.transforms.ResampleTransform attribute) module src.FRAME_FM src.FRAME_FM.cli src.FRAME_FM.dataloaders src.FRAME_FM.dataloaders.cog_dataloader src.FRAME_FM.dataloaders.combined_dataloader src.FRAME_FM.dataloaders.demo_eurosat src.FRAME_FM.dataloaders.demo_geotiff_dataloader src.FRAME_FM.dataloaders.gridded_txt_dataloader src.FRAME_FM.dataloaders.xarray_dataloader src.FRAME_FM.datasets src.FRAME_FM.datasets.base_dataset src.FRAME_FM.datasets.base_gridded_dataset src.FRAME_FM.datasets.base_shapefile_dataset src.FRAME_FM.datasets.chessmet_dataset src.FRAME_FM.datasets.era5_dataset src.FRAME_FM.datasets.ImageLabel_Dataset src.FRAME_FM.datasets.InputOnly_Dataset src.FRAME_FM.datasets.land_cover_map_dataset src.FRAME_FM.datasets.soil_water_index_dataset src.FRAME_FM.datasets.topsoil_dataset src.FRAME_FM.models src.FRAME_FM.models.demo_autoencoder src.FRAME_FM.models.demo_convAE src.FRAME_FM.models.demo_simple_conv src.FRAME_FM.models.mmmae src.FRAME_FM.training src.FRAME_FM.training.logger src.FRAME_FM.training.logging_utils src.FRAME_FM.training.train src.FRAME_FM.transforms src.FRAME_FM.transforms.transforms src.FRAME_FM.utils src.FRAME_FM.utils.data_utils src.FRAME_FM.utils.embedders src.FRAME_FM.utils.LightningDataModuleWrapper src.FRAME_FM.utils.LightningModuleWrapper src.FRAME_FM.utils.settings src.FRAME_FM.utils.transform_utils

MultimodalMaskedAutoencoder (class in src.FRAME_FM.models.mmmae) MyModel (class in src.FRAME_FM.models.demo_simple_conv)

N

n_channels (src.FRAME_FM.utils.embedders.PatchEmbed attribute) n_patches (src.FRAME_FM.utils.embedders.BaseEmbedder attribute) n_variables (in module src.FRAME_FM.models.demo_simple_conv) new_dim (src.FRAME_FM.transforms.transforms.VarsToDimensionTransform attribute) (src.FRAME_FM.transforms.VarsToDimensionTransform attribute) new_name (src.FRAME_FM.transforms.RenameTransform attribute) (src.FRAME_FM.transforms.transforms.RenameTransform attribute)

norm (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) norm_token_loss (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder attribute) NormalizeTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) num_workers (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute)

O

on_validation_epoch_end() (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method) on_validation_epoch_start() (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method)

only_vars_with_time (src.FRAME_FM.transforms.transforms.VarsToDimensionTransform attribute) (src.FRAME_FM.transforms.VarsToDimensionTransform attribute)

P

partition_embed_dim() (in module src.FRAME_FM.utils.embedders) PatchEmbed (class in src.FRAME_FM.utils.embedders) persistent_workers (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) pin_memory (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) pool (src.FRAME_FM.models.demo_simple_conv.SpatialCollapse attribute) pos_embed_ratio (src.FRAME_FM.utils.embedders.STPatchEmbed attribute)

position_space (src.FRAME_FM.utils.embedders.STPatchEmbed attribute) precache (src.FRAME_FM.utils.settings.DatasetSettings attribute) precache_data() (src.FRAME_FM.datasets.base_dataset.BaseDataset method) prepare_data() (src.FRAME_FM.dataloaders.demo_eurosat.EuroSATDataModule method) preprocessor_hash_key (src.FRAME_FM.utils.settings.DatasetSettings attribute) preprocessors (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) proj (src.FRAME_FM.utils.embedders.PatchEmbed attribute)

R

random_masking() (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) reconstruct_dim (src.FRAME_FM.utils.embedders.BaseEmbedder attribute) (src.FRAME_FM.utils.embedders.PatchEmbed attribute) reconstruct_layer (src.FRAME_FM.utils.embedders.PatchEmbed attribute) reconstruct_tokens() (src.FRAME_FM.utils.embedders.BaseEmbedder method) (src.FRAME_FM.utils.embedders.PatchEmbed method) reconstructed_tile_buffer (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) RenameTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms)

ResampleTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) ReshapeTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) resolve_transform() (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms) ReverseAxisTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) RollTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms)

S

safely_remove_dir() (in module src.FRAME_FM.utils.data_utils) SAMPLE_TOPSOIL_CARBON_DATA_URI (in module src.FRAME_FM.datasets.topsoil_dataset) SAMPLE_TOPSOIL_PROPERTIES_DATA_URI (in module src.FRAME_FM.datasets.topsoil_dataset) ScaleTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) setup() (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule method) shape (src.FRAME_FM.transforms.ReshapeTransform attribute) (src.FRAME_FM.transforms.transforms.ReshapeTransform attribute) shift (src.FRAME_FM.transforms.RollTransform attribute) (src.FRAME_FM.transforms.transforms.RollTransform attribute) sincos_embed_coords() (in module src.FRAME_FM.utils.embedders) SoilWaterIndexGriddedTimeSeriesDataset (class in src.FRAME_FM.datasets.soil_water_index_dataset) SortAxisTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) SpatialCollapse (class in src.FRAME_FM.models.demo_simple_conv) split_strategy (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) SqueezeTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) src.FRAME_FM module src.FRAME_FM.cli module src.FRAME_FM.dataloaders module src.FRAME_FM.dataloaders.cog_dataloader module src.FRAME_FM.dataloaders.combined_dataloader module src.FRAME_FM.dataloaders.demo_eurosat module src.FRAME_FM.dataloaders.demo_geotiff_dataloader module src.FRAME_FM.dataloaders.gridded_txt_dataloader module src.FRAME_FM.dataloaders.xarray_dataloader module src.FRAME_FM.datasets module src.FRAME_FM.datasets.base_dataset module src.FRAME_FM.datasets.base_gridded_dataset module src.FRAME_FM.datasets.base_shapefile_dataset module src.FRAME_FM.datasets.chessmet_dataset module src.FRAME_FM.datasets.era5_dataset module src.FRAME_FM.datasets.ImageLabel_Dataset module src.FRAME_FM.datasets.InputOnly_Dataset module

src.FRAME_FM.datasets.land_cover_map_dataset module src.FRAME_FM.datasets.soil_water_index_dataset module src.FRAME_FM.datasets.topsoil_dataset module src.FRAME_FM.models module src.FRAME_FM.models.demo_autoencoder module src.FRAME_FM.models.demo_convAE module src.FRAME_FM.models.demo_simple_conv module src.FRAME_FM.models.mmmae module src.FRAME_FM.training module src.FRAME_FM.training.logger module src.FRAME_FM.training.logging_utils module src.FRAME_FM.training.train module src.FRAME_FM.transforms module src.FRAME_FM.transforms.transforms module src.FRAME_FM.utils module src.FRAME_FM.utils.data_utils module src.FRAME_FM.utils.embedders module src.FRAME_FM.utils.LightningDataModuleWrapper module src.FRAME_FM.utils.LightningModuleWrapper module src.FRAME_FM.utils.settings module src.FRAME_FM.utils.transform_utils module std (src.FRAME_FM.transforms.NormalizeTransform attribute) (src.FRAME_FM.transforms.transforms.NormalizeTransform attribute) STPatchEmbed (class in src.FRAME_FM.utils.embedders) strategy (src.FRAME_FM.transforms.FillMissingValueTransform attribute) (src.FRAME_FM.transforms.transforms.FillMissingValueTransform attribute) subset_selectors (src.FRAME_FM.transforms.SubsetTransform attribute) (src.FRAME_FM.transforms.transforms.SubsetTransform attribute) SubsetTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms)

T

TabularDataset (class in src.FRAME_FM.dataloaders.gridded_txt_dataloader) (class in src.FRAME_FM.dataloaders.xarray_dataloader) target_col (src.FRAME_FM.dataloaders.gridded_txt_dataloader.GriddedDataModule attribute) (src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule attribute) test_dataloader() (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule method) test_dataset (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule attribute) (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) test_fraction (src.FRAME_FM.dataloaders.gridded_txt_dataloader.GriddedDataModule attribute) (src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule attribute) test_split (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) test_step() (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) test_step_body() (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder method) (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) test_transforms (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) tile_array() (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule method) tile_size (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule attribute) tile_sizes (src.FRAME_FM.transforms.TilerTransform attribute) (src.FRAME_FM.transforms.transforms.TilerTransform attribute) TilerTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) time_stride (src.FRAME_FM.datasets.base_gridded_dataset.BaseGriddedTimeSeriesDataset attribute) to_latent (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder attribute) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder attribute) ToDataArray (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) tokenify() (src.FRAME_FM.utils.embedders.BaseEmbedder method) (src.FRAME_FM.utils.embedders.PatchEmbed method) (src.FRAME_FM.utils.embedders.STPatchEmbed method)

TopsoilDataset (class in src.FRAME_FM.datasets.topsoil_dataset) ToTensorTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) train_dataloader() (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule method) train_dataset (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule attribute) (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) train_split (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) train_transforms (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) training_step() (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) training_step_body() (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoderWithLocation method) (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) transform (src.FRAME_FM.datasets.ImageLabel_Dataset.TransformedDataset attribute) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputCoordsDataset attribute) (src.FRAME_FM.datasets.InputOnly_Dataset.TransformedInputDataset attribute) transform_mapping (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms) TransformedDataset (class in src.FRAME_FM.datasets.ImageLabel_Dataset) TransformedInputCoordsDataset (class in src.FRAME_FM.datasets.InputOnly_Dataset) TransformedInputDataset (class in src.FRAME_FM.datasets.InputOnly_Dataset) transforms (src.FRAME_FM.datasets.base_dataset.BaseDataset attribute) (src.FRAME_FM.datasets.base_shapefile_dataset.BaseShapefileDataset attribute) TransposeTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms) TT (in module src.FRAME_FM.transforms) (in module src.FRAME_FM.transforms.transforms)

U

unify_transforms() (in module src.FRAME_FM.utils.data_utils)

untokenify() (src.FRAME_FM.utils.embedders.BaseEmbedder method) (src.FRAME_FM.utils.embedders.PatchEmbed method)

V

val_dataloader() (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule method) val_dataset (src.FRAME_FM.dataloaders.demo_geotiff_dataloader.XarrayStaticDataModule attribute) (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) val_fraction (src.FRAME_FM.dataloaders.gridded_txt_dataloader.GriddedDataModule attribute) (src.FRAME_FM.dataloaders.xarray_dataloader.GriddedDataModule attribute) val_split (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) val_transforms (src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule attribute) validation_step() (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method) validation_step_body() (src.FRAME_FM.models.demo_autoencoder.EuroSATAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoder method) (src.FRAME_FM.models.demo_convAE.ConvAutoencoderWithLocation method) (src.FRAME_FM.models.mmmae.MultimodalMaskedAutoencoder method) (src.FRAME_FM.utils.LightningModuleWrapper.BaseModule method)

var_id (src.FRAME_FM.transforms.RenameTransform attribute) (src.FRAME_FM.transforms.ToDataArray attribute) (src.FRAME_FM.transforms.transforms.RenameTransform attribute) (src.FRAME_FM.transforms.transforms.ToDataArray attribute) variables (src.FRAME_FM.transforms.transforms.VarsToDimensionTransform attribute) (src.FRAME_FM.transforms.VarsToDimensionTransform attribute) VarsToDimensionTransform (class in src.FRAME_FM.transforms) (class in src.FRAME_FM.transforms.transforms)

W

write_zarr() (in module src.FRAME_FM.utils.data_utils)

X

XarrayStaticDataModule (class in src.FRAME_FM.dataloaders.demo_geotiff_dataloader)

Z

zarr_format (src.FRAME_FM.utils.settings.DefaultSettings attribute)