Skip to content

Commit

Permalink
raise an exception for stratified split
Browse files Browse the repository at this point in the history
  • Loading branch information
sdpython committed Sep 24, 2017
1 parent f3fa2cc commit 7ef9139
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/pandas_streaming/df/dataframe.py
Expand Up @@ -258,6 +258,9 @@ def train_test_split(self, path_or_buf=None, export_method="to_csv",
the same iterator again as random draws would be different.
We need to store the results into files or buffers.
"""
if kwargs.get("stratify") is not None:
raise NotImplementedError(
"No implementation yet for the stratified version.")
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=ImportWarning)
from sklearn.model_selection import train_test_split
Expand Down

0 comments on commit 7ef9139

Please sign in to comment.