From ed78e65dd8492c17edbb72f04c93bd0917cd5c9d Mon Sep 17 00:00:00 2001 From: jajupmochi Date: Fri, 24 Jun 2022 17:48:16 +0200 Subject: [PATCH] [Features] Add model seletion methods with validation set: KFoldWithValid, RepeatedKFoldWithValid, BaseCrossValidatorWithValid. Required version of scikit-learn is upgraded to 1.1.0, to support the input_name argument of sklearn.utils.check_array used in gklearn.model_selection._split.py. --- gklearn/model_selection/_split.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/gklearn/model_selection/_split.py b/gklearn/model_selection/_split.py index a982fec..10d5cc8 100644 --- a/gklearn/model_selection/_split.py +++ b/gklearn/model_selection/_split.py @@ -4,6 +4,8 @@ Created on Fri Jun 24 11:13:26 2022 @author: ljia + +Reference: scikit-learn. """ from abc import abstractmethod import numbers @@ -85,7 +87,7 @@ class BaseCrossValidatorWithValid(object): class _BaseKFoldWithValid(BaseCrossValidatorWithValid): - """Base class for KFold, GroupKFold, and StratifiedKFold""" + """Base class for KFoldWithValid, GroupKFoldWithValid, and StratifiedKFoldWithValid""" @abstractmethod def __init__(self, n_splits, *, stratify, shuffle, random_state):