Skip to main content

GridSearchCV

#include <Skigen/ModelSelection>

template <typename Estimator, typename Scalar = double>
class Skigen::GridSearchCV(estimator, param_grid, cv=5, refit=true, n_jobs=1)

Exhaustive search over a parameter grid with cross-validation.

Mirrors sklearn.model_selection.GridSearchCV.

Fits the estimator for every point in the Cartesian product of the supplied parameter grid, evaluates each via K-fold cross-validation, and optionally refits the best configuration on the full training set.



Attributes:

  • best_estimator : Estimator

  • best_params : ParameterDict

  • best_score : Scalar

  • best_index : std::size_t

  • cv_results_params : const std::vector< ParameterDict > &

  • cv_results_mean_score : const std::vector< Scalar > &


Methods

fit(X, y)


predict(X)


score(X, y)