grid_search

A simple grid search over specified parameter values for a model.

Note. After searching, the model is trained and ready to use.

Method call format

grid_search(param_grid,
            X,
            y=None,
            cv=3,
            partition_random_seed=0,
            calc_cv_statistics=True,
            search_by_train_test_split=True,
            refit=True,
            shuffle=True,
            stratified=None,
            train_size=0.8,
            verbose=True,
            plot=False)

Parameters

Parameter Possible types Description Default value

param_grid

  • dict
  • list

Dictionary with parameters names (string) as keys and lists of parameter settings to try as values, or a list of such dictionaries, in which case the grids spanned by each dictionary in the list are explored.

This enables searching over any sequence of parameter settings.

Required parameter

X

catboost.Pool