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

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
)

Description

Create a version of a model.

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

Examples

model_v0 = model.create_version()

Returns

A ModelVersion object

get_version

get_version(
   version: int
)

Description

Retrieve a version of a model

Examples

version = model.get_version(0)

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