def _check_cv_float(self): cv_cls = StratifiedShuffleSplit if self.stratified else ShuffleSplit return cv_cls(test_size=self.cv, random_state=self.random_state)