Browse Source

[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.

v0.2.x
jajupmochi 3 years ago
parent
commit
ed78e65dd8
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      gklearn/model_selection/_split.py

+ 3
- 1
gklearn/model_selection/_split.py View File

@@ -4,6 +4,8 @@
Created on Fri Jun 24 11:13:26 2022 Created on Fri Jun 24 11:13:26 2022


@author: ljia @author: ljia

Reference: scikit-learn.
""" """
from abc import abstractmethod from abc import abstractmethod
import numbers import numbers
@@ -85,7 +87,7 @@ class BaseCrossValidatorWithValid(object):




class _BaseKFoldWithValid(BaseCrossValidatorWithValid): class _BaseKFoldWithValid(BaseCrossValidatorWithValid):
"""Base class for KFold, GroupKFold, and StratifiedKFold"""
"""Base class for KFoldWithValid, GroupKFoldWithValid, and StratifiedKFoldWithValid"""


@abstractmethod @abstractmethod
def __init__(self, n_splits, *, stratify, shuffle, random_state): def __init__(self, n_splits, *, stratify, shuffle, random_state):


Loading…
Cancel
Save