Skip to content

Latest commit

 

History

History
37 lines (25 loc) · 922 Bytes

model_selection.rst

File metadata and controls

37 lines (25 loc) · 922 Bytes

The model_selection package

Surprise provides various tools to run cross-validation procedures and search the best parameters for a prediction algorithm. The tools presented here are all heavily inspired from the excellent scikit learn library.

Cross validation iterators

.. automodule:: surprise.model_selection.split
    :members:
    :exclude-members: get_cv, get_rng

Cross validation

.. autofunction:: surprise.model_selection.validation.cross_validate

Parameter search

.. autoclass:: surprise.model_selection.search.GridSearchCV
    :members:
    :inherited-members:

.. autoclass:: surprise.model_selection.search.RandomizedSearchCV
    :members:
    :inherited-members: