Loading...

vastorbit.machine_learning.vast.automl.AutoML.fit

AutoML.fit(input_relation: Annotated[str | VastFrame, ''], X: Annotated[str | list[str], 'STRING representing one column or a list of columns'] | None = None, y: str | None = None, return_report: bool = False) None

Trains the model.

Parameters:
  • input_relation (SQLRelation) – Training Relation.

  • X (SQLColumns, optional) – List of the predictors.

  • y (str, optional) – Response column.