lib/svmkit/model_selection/k_fold.rb in svmkit-0.2.8 vs lib/svmkit/model_selection/k_fold.rb in svmkit-0.2.9

@@ -33,10 +33,10 @@
       # @param random_seed [Integer] The seed value using to initialize the random generator.
       def initialize(n_splits: 3, shuffle: false, random_seed: nil)
         SVMKit::Validation.check_params_integer(n_splits: n_splits)
         SVMKit::Validation.check_params_boolean(shuffle: shuffle)
         SVMKit::Validation.check_params_type_or_nil(Integer, random_seed: random_seed)
-
+        SVMKit::Validation.check_params_positive(n_splits: n_splits)
         @n_splits = n_splits
         @shuffle = shuffle
         @random_seed = random_seed
         @random_seed ||= srand
         @rng = Random.new(@random_seed)