Skip to content

Latest commit

 

History

History
14 lines (9 loc) · 670 Bytes

model.md

File metadata and controls

14 lines (9 loc) · 670 Bytes

Model

Model represents a machine learning model. Each Model has a type, currently Merlin supports both standard model (PyTorch, SKLearn, Tensorflow, and XGBoost) and user-defined model (PyFunc model).

Conceptually, Model in Merlin is similar to a class in programming language. To instantiate a Model you’ll have to create a Model Version.

merlin.set_model(<model_name>, <model_type>) will set the active model to the name given by parameter. If the Model with given name is not found, a new Model will be created.

import merlin
from merlin.model import ModelType

merlin.set_model("tensorflow-model", ModelType.TENSORFLOW)