Source code for steamship.plugin.inputs.train_plugin_input

from __future__ import annotations

from typing import Optional

from pydantic import Field

from steamship.base.model import CamelModel


[docs] class TrainPluginInput(CamelModel): """ This is the object passed as input to a trainable operation, stored as the `input` field of a `train` task. """ plugin_instance: str # How may epochs of trainable to perform, if relevant and supported training_epochs: Optional[int] = None # How much data to hold out for testing & reporting, if relevant and supported. testing_holdout_percent: Optional[float] = None # An optional seed for the train-test split test_split_seed: Optional[int] = None # Arbitrary key-valued data to provide to the particular `modelName` trainer. training_params: Optional[dict] = None # Arbitrary key-valued data to provide to the inference runner in the TrainPluginOutput object. # The trainable process will have the opportunity to amend this before writing it to the output inference_params: Optional[dict] = None # A pre-signed URL at which the trainable data can be found training_data_url: Optional[str] = Field(None, alias="trainingDataUrl")