/
githubmirror
/
scikit-learn
Обзор
Документация
Войти
/
githubmirror
/
scikit-learn
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
sklearn/model_selection/tests/common.py
24 строки
641 B
Thomas J. Fan
MNT Applies black formatting to most of the code base (#18948)
17 июн 2021, 21:21
Не верифицирован
17 июн 2021, 21:21
82df489
Код
Авторство
О чём код?
""" Common utilities for testing model selection. """ import numpy as np from sklearn.model_selection import KFold class OneTimeSplitter: """A wrapper to make KFold single entry cv iterator""" def __init__(self, n_splits=4, n_samples=99): self.n_splits = n_splits self.n_samples = n_samples self.indices = iter(KFold(n_splits=n_splits).split(np.ones(n_samples))) def split(self, X=None, y=None, groups=None): """Split can be called only once""" for index in self.indices: yield index def get_n_splits(self, X=None, y=None, groups=None): return self.n_splits