Package ai.djl
Contains top level, engine-agnostic classes for both inference and training.
-
Interface Summary Interface Description Application.CV The common set of applications for computer vision (image and video data).Application.NLP The common set of applications for natural language processing (text data).Application.Tabular The common set of applications for tabular data.Device.Type Contains device type string constants.DeviceType DeviceType
is a class used to map the Device name to their corresponding type number.Model A model is a collection of artifacts that is created by the training process. -
Class Summary Class Description Application A class contains common tasks that can be completed using deep learning.BaseModel BaseModel
is the basic implementation ofModel
.Device TheDevice
class provides the specified assignment for CPU/GPU processing on theNDArray
. -
Exception Summary Exception Description MalformedModelException Thrown to indicate Model parameters are not in expected format or are malformed.ModelException Thrown to indicate Model parameter or load exceptions parent to Model Exceptions.TrainingDivergedException Thrown to indicate when there is a divergence during Training.