Model

Properties

  • name Name of this Model

  • type Type of this Model

  • framework Framework of this Model

  • private Privacy of this Model


Methods

add_tags

add_tags(
   tags: Union[Tag, List[Tag]]
)

Description

Add some tags to an object.
It can be used on Data/MultiData/Asset/MultiAsset/DatasetVersion/Dataset/Model/ModelVersion.

You can give a Tag or a list of Tag.

Examples

tag_bicycle = client.create_tag("bicycle", Target.DATA)
tag_car = client.create_tag("car", Target.DATA)
tag_truck = client.create_tag("truck", Target.DATA)

data.add_tags(tag_bicycle)
data.add_tags([tag_car, tag_truck])

update

update(
   name: Optional[str] = None, framework: Union[str, Framework, None] = None,
   private: Optional[bool] = None, description: Optional[str] = None, type: Union[str,
   InferenceType, None] = None
)

Description

Update a model with a new name, framework, privacy, description or type

Arguments

  • name (str) : Model name to create.

  • description (str) : Description of this model

Examples

model.update(description="Very cool model")

delete

delete()

Description

Delete model.

Delete the model in Picsellia database

Examples

model.delete()

get_tags

get_tags()

Description

Retrieve the tags of your model.

Examples

tags = my_model.get_tags()
assert tags[0].name == "my-model-1"

Returns

List of tags as Tag


create_version

create_version(
   docker_image_name: Optional[str] = None, docker_flags: Optional[List[str]] = None,
   thumb_object_name: Optional[str] = None, notebook_link: Optional[str] = None,
   labels: Optional[dict] = None, base_parameters: Optional[dict] = None,
   docker_env_variables: Optional[dict] = None, name: Optional[str] = None,
   framework: Union[str, Framework, None] = None, type: Union[str, InferenceType,
   None] = None, description: Optional[str] = None, docker_tag: Optional[str] = None
)

Description

Create a version of a model.

The version number of this model will be defined by the platform. It is incremented automatically.

Examples

Returns

A ModelVersion object


get_version

get_version(
   version: Union[int, str]
)

Description

Retrieve a version of a model from its version or its name

Examples

# Assuming model is a Model without version
model_version_a = model.create_version("first-version")
model_version_b = model.get_version(0)
model_version_c = model.get_version("first-version")
assert model_version_a == model_version_b
assert model_version_a == model_version_c

Returns

A ModelVersion object


list_versions

list_versions(
   limit: Optional[int] = None, offset: Optional[int] = None,
   order_by: Optional[List[str]] = None
)

Description

List versions of a model

Examples

versions = model.list_versions()

Returns

A list of ModelVersion object of this model