Add warning that shufflesplits unsupported

1 files changed, 18 insertions, 1 deletions

diff --git a/gensvm/gridsearch.py b/gensvm/gridsearch.py
index 7d453c9..0eeb603 100644
--- a/gensvm/gridsearch.py
+++ b/gensvm/gridsearch.py
@@ -17,7 +17,12 @@ import time
 from operator import itemgetter
 
 from sklearn.base import ClassifierMixin, BaseEstimator, MetaEstimatorMixin
-from sklearn.model_selection import ParameterGrid, check_cv
+from sklearn.model_selection import (
+    ParameterGrid,
+    check_cv,
+    ShuffleSplit,
+    StratifiedShuffleSplit,
+)
 from sklearn.model_selection._search import _check_param_grid
 from sklearn.model_selection._validation import _score
 from sklearn.preprocessing import LabelEncoder
@@ -328,6 +333,12 @@ class GenSVMGridSearchCV(BaseEstimator, MetaEstimatorMixin):
         Refer to the `scikit-learn User Guide on cross validation`_ for the 
         various strategies that can be used here.
 
+        NOTE: At the moment, the ShuffleSplit and StratifiedShuffleSplit are 
+        not supported in this class. If you need these, you can use the GenSVM 
+        classifier directly with the GridSearchCV object from scikit-learn.  
+        (these methods require significant changes in the low-level library 
+        before they can be supported).
+
     refit : boolean, or string, default=True
         Refit the GenSVM estimator with the best found parameters on the whole 
         dataset.
@@ -520,6 +531,12 @@ class GenSVMGridSearchCV(BaseEstimator, MetaEstimatorMixin):
 
         self.scoring = scoring
         self.cv = 5 if cv is None else cv
+        if isinstance(self.cv, ShuffleSplit) or isinstance(
+            self.cv, StratifiedShuffleSplit
+        ):
+            raise ValueError(
+                "ShuffleSplit and StratifiedShuffleSplit are not supported at the moment. Please see the documentation for more info"
+            )
         self.refit = refit
         self.verbose = verbose
         self.return_train_score = return_train_score