Skip to main content

Models

Models are the core of Predibase. With the SDK you can train models, predict with models, deploy models, run queries with models, and much more.

Attributes

id    ID of the model

config    Config that the model was trained with

version The version of this model

repo_id The ID of the model repo that this model belongs to

repo    The repo object that this model belongs to

dataset_id     The ID of the dataset that this model was trained on

dataset   The dataset object that this model was trained on

status     The current status of the model (training, deployed, errored, etc.)

created   Time of model creation

completed     Time of completion whether that be train completion, error, or cancellation

error_text The error message describing what went wrong if the model failed

Methods

Model.delete

Model.fork

Model.predict

Model.to_draft