Retrieve TrainingSets

You can get TrainingSets using their version number or using an explicit search.

Get TrainingSetVersion

TrainingSetVersions can be retrieved by name and number.

tsv_by_num = TrainingSetClient.get_training_set_version(
    training_set_name="my-training-set",
    number=2,
)

You can also access the raw data from the TrainingSetVersion.

raw_df = tsv_by_num.load_raw_pandas()

Training data can also be accessed.

training_df = tsv_by_num.load_training_pandas()

Find TrainingSetVersions

You can search for TrainingSetVersions for the current project using the training set name, TrainingSet metadata, and TrainingSetVersion metadata. Results matching all search fields are returned.

versions = TrainingSetClient.list_training_set_versions(
    training_set_name="my-training-set",
    meta={"year": "2021"},
    training_set_meta={"category": "widgets"}
)

Get TrainingSet

You can retrieve TrainingSets by name.

ts = TrainingSetClient.get_training_set("my-training-set")

Find TrainingSets

You can search TrainingSets by metadata. Results matching all metadata fields are returned.

ts = TrainingSetClient.list_training_sets(
    meta={"category": "widgets"},
)