Loading...

vastorbit.machine_learning.vast.pipeline.Pipeline.fit

Pipeline.fit(input_relation: Annotated[str | VastFrame, ''], X: list, y: str | None = None, test_relation: Annotated[str | VastFrame, ''] = '', return_report: bool = False) None

Trains the model.

Parameters:
  • input_relation (SQLRelation) – Training relation.

  • X (list) – List of the predictors.

  • y (str, optional) – Response column.

  • test_relation (SQLRelation, optional) – Relation used to test the model.

Returns:

model.

Return type:

object