Index

_

__init__() (unlocknn.kernel_layers.AmpAndLengthScaleFn method) (unlocknn.kernel_layers.KernelLayer method)

_amplitude_basis (unlocknn.kernel_layers.AmpAndLengthScaleFn attribute) _length_scale_basis (unlocknn.kernel_layers.AmpAndLengthScaleFn attribute)

A

AmpAndLengthScaleFn (class in unlocknn.kernel_layers)

amplitude (unlocknn.kernel_layers.AmpAndLengthScaleFn property) AVAILABLE_METRICS (in module unlocknn.metrics)

C

call() (unlocknn.kernel_layers.KernelLayer method) ckpt_callback() (unlocknn.model.ProbNN method) compile() (unlocknn.model.ProbNN method)

config (unlocknn.kernel_layers.KernelLayer property) (unlocknn.model.ProbNN property) CONFIG_VARS (unlocknn.model.ProbNN attribute) create_megnet_input() (in module unlocknn.megnet_utils)

D

default_megnet_config() (in module unlocknn.megnet_utils)

E

evaluate() (unlocknn.model.MEGNetProbModel method)

evaluate_uq_metrics() (in module unlocknn.metrics)

K

kernel (unlocknn.kernel_layers.KernelLayer property)

KernelLayer (class in unlocknn.kernel_layers)

L

length_scale (unlocknn.kernel_layers.AmpAndLengthScaleFn property) load() (unlocknn.model.MEGNetProbModel class method) (unlocknn.model.ProbNN class method)

load_data() (in module unlocknn.download) load_kernel() (in module unlocknn.kernel_layers) load_pretrained() (in module unlocknn.download)

M

MEGNetProbModel (class in unlocknn.model)

module unlocknn

N

neg_log_likelihood() (in module unlocknn.metrics) nn_frozen (unlocknn.model.MEGNetProbModel property) (unlocknn.model.ProbNN property)

norm_frozen (unlocknn.model.MEGNetProbModel property) (unlocknn.model.ProbNN property)

P

predict() (unlocknn.model.MEGNetProbModel method) (unlocknn.model.ProbNN method)

ProbNN (class in unlocknn.model)

S

SampleInitializer (class in unlocknn.initializers) save() (unlocknn.kernel_layers.KernelLayer method) (unlocknn.model.MEGNetProbModel method) (unlocknn.model.ProbNN method) save_kernel() (unlocknn.model.ProbNN method)

save_struct_data() (in module unlocknn.download) scale_targets() (in module unlocknn.megnet_utils) set_frozen() (unlocknn.model.MEGNetProbModel method) (unlocknn.model.ProbNN method) sharpness() (in module unlocknn.metrics)

T

train() (unlocknn.model.MEGNetProbModel method) (unlocknn.model.ProbNN method)

U

unlocknn module

update_pred_model() (unlocknn.model.ProbNN method)

V

variation() (in module unlocknn.metrics)

vgp_frozen (unlocknn.model.MEGNetProbModel property) (unlocknn.model.ProbNN property)