Index

A

add_cutout() (in module edafm.preprocessing) add_gradient() (in module edafm.preprocessing) add_losses() (edafm.utils.LossLogPlot method)

add_noise() (in module edafm.preprocessing) add_norm() (in module edafm.preprocessing) add_rotation_reflection() (in module edafm.preprocessing) AttentionUNet (class in edafm.models)

C

Conv2dBlock (class in edafm.layers)

Conv3dBlock (class in edafm.layers) count_parameters() (in module edafm.utils)

D

download_molecules() (in module edafm.utils)

download_weights() (in module edafm.utils)

E

edafm.layers module edafm.models module edafm.preprocessing module

edafm.training module edafm.utils module edafm.visualization module EDAFMNet (class in edafm.models)

F

forward() (edafm.layers.UNetAttentionConv method) (edafm.models.AttentionUNet method) (edafm.training.ImgLoss method)

I

ImgDataset (class in edafm.training)

ImgLoss (class in edafm.training) interpolate_and_crop() (in module edafm.preprocessing)

L

load_checkpoint() (in module edafm.utils)

LossLogPlot (class in edafm.utils)

M

make_dataloader() (in module edafm.training) make_input_plots() (in module edafm.visualization) make_prediction_plots() (in module edafm.visualization) module edafm.layers edafm.models edafm.preprocessing edafm.training edafm.utils edafm.visualization

P

plot_history() (edafm.utils.LossLogPlot method)

plot_input() (in module edafm.visualization)

R

rand_shift_xy_trend() (in module edafm.preprocessing)

random_crop() (in module edafm.preprocessing) read_xyzs() (in module edafm.utils)

S

save_checkpoint() (in module edafm.utils)

shuffle() (edafm.training.ImgDataset method)

T

training (edafm.layers.Conv2dBlock attribute) (edafm.layers.Conv3dBlock attribute) (edafm.layers.UNetAttentionConv attribute) (edafm.models.AttentionUNet attribute) (edafm.models.EDAFMNet attribute) (edafm.training.ImgLoss attribute)

U

UNetAttentionConv (class in edafm.layers)