Skip to content

Commit

Permalink
refactor(frontend): remove get_n_split from StratifiedKFold and leave…
Browse files Browse the repository at this point in the history
… the call to the base class KFold in sklearn front
  • Loading branch information
Ishticode committed Sep 1, 2023
1 parent 1eb64d5 commit f78cb3f
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions ivy/functional/frontends/sklearn/model_selection/_split.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,9 +75,6 @@ def __init__(
def _iter_test_indices(self, X=None, y=None, groups=None):
raise NotImplementedError

def get_n_splits(self, X=None, y=None, groups=None):
return self.n_splits


@to_ivy_arrays_and_back
def train_test_split(
Expand Down

0 comments on commit f78cb3f

Please sign in to comment.