kumoai.pquery.TrainingTable

class kumoai.pquery.TrainingTable

Bases: object

A training table in the Kumo platform. A training table can be initialized from a job ID of a completed training table generation job.

import kumoai

# Create a Training Table from a training table generation job. Note
# that the job ID passed here must be in a completed state:
training_table = kumoai.TrainingTable("gen-traintable-job-...")

# Read the training table as a Pandas DataFrame:
training_df = training_table.data_df()

# Get URLs to download the training table:
training_download_urls = training_table.data_urls()

Parameters:

job_id (str) – ID of the training table generation job which generated this training table.

__init__(job_id)

data_urls()

Returns a list of URLs that can be used to view generated training table data. The list will contain more than one element if the table is partitioned; paths will be relative to the location of the Kumo data plane.

Return type:

List[str]

data_df()

Returns a DataFrame object representing the generated training data. :rtype: DataFrame

Warning

This method will load the full training table into memory as a DataFrame object. If you are working on a machine with limited resources, please use data_urls() instead to download the data and perform analysis per-partition.