AveragingTransformer

class sparklightautoml.ml_algo.base.AveragingTransformer(task_name=None, input_cols=None, output_col='averaged_values', remove_cols=None, convert_to_array_first=False, weights=None, dim_num=1)

Bases: Transformer, HasInputCols, HasOutputCol, DefaultParamsWritable, DefaultParamsReadable

Transformer that gets one or more columns and produce column with average values.

__init__(task_name=None, input_cols=None, output_col='averaged_values', remove_cols=None, convert_to_array_first=False, weights=None, dim_num=1)

Parameters:

• task_name (str, optional) – Task name: “binary”, “multiclass” or “reg”.

• input_cols (List[str], optional) – List of input columns.

• output_col (str, optional) – Output column name. Defaults to “averaged_values”.

• remove_cols (Optional[List[str]], optional) – Columns need to remove. Defaults to None.

• convert_to_array_first (bool, optional) – If True then will be convert input vectors to arrays. Defaults to False.

• weights (Optional[List[int]], optional) – List of weights to scaling output values. Defaults to None.

• dim_num (int, optional) – Dimension of input columns. Defaults to 1.