kumoai.trainer.ModelPlan#

class kumoai.trainer.ModelPlan[source]#

A complete definition of a Kumo model plan, encompassing a TrainingJobPlan, ColumnProcessingPlan, NeighborSamplingPlan, OptimizationPlan, and a ModelArchitecturePlan. Please see the Kumo documentation for more information.

training_job: TrainingJobPlan#

The training job plan.

neighbor_sampling: NeighborSamplingPlan#

The neighbor sampling plan.

optimization: OptimizationPlan#

The model optimization plan.

model_architecture: ModelArchitecturePlan#

The model architecture plan.