atomai_models.rst

AtomAI Models

Segmentor

atomai.models.Segmentor

ImSpec

atomai.models.ImSpec

Variational Autoencoder (VAE)

atomai.models.VAE

Rotational Variational Autoencoder (rVAE)

atomai.models.rVAE

Joint Variational Autoencoder (jVAE)

atomai.models.jVAE

Joint Rotational Variational Autoencoder (jrVAE)

atomai.models.jrVAE

Deep Kernel Learning

atomai.models.dklGPR

Load trained models

atomai.models.load_model

atomai.models.load_ensemble