MLModel

abstract class MLModel

The MLModel class abstracts a self-contained machine learning model, containing a computation graph for making predictions along with feature type information and metadata.

circle-info

Models are created from MLModelData instances.

NatML offers two types of models depending on where predictions are made:

circle-info

Currently, the Node SDK only supports Hub models.

Inspecting Feature Types

Models provide information about their expected input and output feature types. This type information is crucial for writing some model predictors.

Input Features

/**
 * Model input feature types.
 */
inputs: MLFeatureType[];

The model provides its expected input feature types. Typically, a predictor will handle any necessary conversions of your input feature so that it matches the type that the model expects.

circle-info

The inputs are reported in the same order that they are expected by the model when making predictions.

circle-exclamation

Output Features

The model provides its output feature types. This information is crucial in order to convert the model's raw outputs into more usable forms by predictors.

circle-info

The outputs are reported in the same order that they are produced by the model when making predictions.

circle-exclamation

Inspecting Metadata

Models expose metadata that was defined when they were created.

circle-exclamation

Disposing the Model

Models create native and/or server resources. As a result, you must dispose of the model once you are done using it.

triangle-exclamation

Last updated