Skip to content

Commit

Permalink
Spacing docs
Browse files Browse the repository at this point in the history
  • Loading branch information
FrancescMartiEscofetQC committed Jul 5, 2024
1 parent 669f37f commit 7b97173
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions metalearners/grid_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,14 +135,15 @@ class MetaLearnerGridSearch:
Importantly, ``random_state`` must be passed through the ``random_state`` parameter
and not through ``metalearner_params``.
``base_learner_grid`` keys should be the names of the needed base models contained in the :class:`~metalearners.metalearners.MetaLearner`
defined by ``metalearner_factory``, for information about this names check
``base_learner_grid`` keys should be the names of the needed base models contained in the
:class:`~metalearners.metalearners.MetaLearner` defined by ``metalearner_factory``, for
information about this names check
:meth:`~metalearners.metalearner.MetaLearner.nuisance_model_specifications` and
:meth:`~metalearners.metalearner.MetaLearner.treatment_model_specifications`. The
values should be sequences of model factories.
If base models are meant to be reused, they should be passed through ``metalearner_params`` and the corresponding keys
should not be passed to ``base_learner_grid``.
If base models are meant to be reused, they should be passed through ``metalearner_params``
and the corresponding keys should not be passed to ``base_learner_grid``.
``param_grid`` should contain the parameters grid for each type of model used by the
base learners defined in ``base_learner_grid``. The keys should be strings with the
Expand Down

0 comments on commit 7b97173

Please sign in to comment.