Skip to main content

layer.Model

class Model()

Provides access to ML models trained and stored in Layer.

You can retrieve an instance of this object with layer.get_model().

This class should not be initialized by end-users.

# Fetches a specific version of this model
layer.get_model("churn_model:1.2")

get_train()

Returns the trained and saved model artifact. For example, a scikit-learn or PyTorch model object.

  • Returns

    The trained model artifact.

  • Return type

    ModelObject

predict()

Performs prediction on the input dataframe data. :return: the predictions as a pd.DataFrame

  • Parameters

    input_df (pandas.core.frame.DataFrame) --

  • Return type

    pandas.core.frame.DataFrame