[GRASS-SVN] r70959 - grass-addons/grass7/raster/r.learn.ml

[svn_grass at osgeo.org]

Author: spawley
Date: 2017-04-26 10:42:50 -0700 (Wed, 26 Apr 2017)
New Revision: 70959

Modified:
   grass-addons/grass7/raster/r.learn.ml/r.learn.ml.py
Log:
r.learn fixed issue when performing no cross validation

Modified: grass-addons/grass7/raster/r.learn.ml/r.learn.ml.py
===================================================================
--- grass-addons/grass7/raster/r.learn.ml/r.learn.ml.py	2017-04-26 14:29:14 UTC (rev 70958)
+++ grass-addons/grass7/raster/r.learn.ml/r.learn.ml.py	2017-04-26 17:42:50 UTC (rev 70959)
@@ -592,17 +592,18 @@
                 inner = ShuffleSplit(n_splits=1, test_size=0.33, random_state=random_state)
             else:
                 inner = GroupShuffleSplit(n_splits=1, test_size=0.33, random_state=random_state)
-        
+
         else:
             inner = None
 
         # ---------------------------------------------------------------------
         # define the outer search resampling method
         # ---------------------------------------------------------------------
-        if group_id is None:
-            outer = StratifiedKFold(n_splits=cv, random_state=random_state)
-        else:
-            outer = GroupKFold(n_splits=cv)
+        if cv > 1:
+            if group_id is None:
+                outer = StratifiedKFold(n_splits=cv, random_state=random_state)
+            else:
+                outer = GroupKFold(n_splits=cv)
 
         # ---------------------------------------------------------------------
         # define sample weights for gradient boosting classifiers