Put back to 10 classifiers

igrigorik · Apr 11, 2017 · 868ed91 · 868ed91
1 parent ff298a8
commit 868ed91
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
@@ -3,8 +3,6 @@
 ### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com>
 ### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com>
 
-require 'set'
-
 module DecisionTree
   Node = Struct.new(:attribute, :threshold, :gain)
 
@@ -340,7 +338,7 @@ def initialize(attributes, data, default, type)
     end
 
     def train(data = @data, attributes = @attributes, default = @default)
-      @classifiers = 5.times.map do |i|
+      @classifiers = 10.times.map do |i|
         Ruleset.new(attributes, data, default, @type)
       end