Login
Back to bricks list
Introduction Version

KNN regressor trainer

TASK
Typing name :  TASK.gws_gaia.KNNRegressorTrainer Brick :  gws_gaia

Train a k-nearest neighbors regression model

Trainer for a k-nearest neighbors regressor. Fit a k-nearest neighbors regressor from a training table.

See https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsRegressor.html for more details.

Input

Table
The input table

Output

result
The output result

Configuration

training_design

Define the training design, i.e. the target Y to use for the model.

Type : ListMaximum occurrences number : -1

target_name

The name of the 'columns' or 'row_tag keys' to use as targets or labels.

Type : string

target_origin

Optional

The origin of the target. Notice: Targets comming from a 'row_tag' are always considered as categorical.

Type : stringAllowed values : column  row_tag  Default value : column

target_type

Optional

The type of the target (categorical or numerical). Set 'auto' to infer the correct type. Notice: targets comming from row_tags are allways considered as categorical

Type : stringAllowed values : auto  categorical  numerical  Default value : auto

nb_neighbors

Optional

Type : intDefault value : 5