- VAEP.fit(X, y, learner='xgboost', val_size=0.25, tree_params=None, fit_params=None)#
Fit the model according to the given training data.
X (pd.DataFrame) – Feature representation of the game states.
y (pd.DataFrame) – Scoring and conceding labels for each game state.
learner (string, default='xgboost' # noqa: DAR103) – Gradient boosting implementation which should be used to learn the model. The supported learners are ‘xgboost’, ‘catboost’ and ‘lightgbm’.
val_size (float, default=0.25 # noqa: DAR103) – Percentage of the dataset that will be used as the validation set for early stopping. When zero, no validation data will be used.
tree_params (dict) – Parameters passed to the constructor of the learner.
fit_params (dict) – Parameters passed to the fit method of the learner.
ValueError – If one of the features is missing in the provided dataframe.
Fitted VAEP model.
- Return type