lib/eps/naive_bayes.rb in eps-0.3.4 vs lib/eps/naive_bayes.rb in eps-0.3.5

- old
+ new

@@ -15,10 +15,10 @@ str += "#{k}: #{v}\n" end str end - def _train(smoothing: 1, **options) + def _train(smoothing: 1) raise "Target must be strings" if @target_type != "categorical" check_missing_value(@train_set) check_missing_value(@validation_set) if @validation_set raise ArgumentError, "weight not supported" if @train_set.weight