dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-28 07:04:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Put back to 10 classifiers

Browse Source
This commit is contained in:
Brian Underwood
2017-04-11 13:05:13 -04:00
parent ff298a8fb9
commit 868ed91672
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/decisiontree/id3_tree.rb
View File

@@ -3,8 +3,6 @@
### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com> ### Copyright (c) 2007 Ilya Grigorik <ilya AT igvita DOT com>
### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com> ### Modifed at 2007 by José Ignacio Fernández <joseignacio.fernandez AT gmail DOT com>
require 'set'
module DecisionTree module DecisionTree
Node = Struct.new(:attribute, :threshold, :gain) Node = Struct.new(:attribute, :threshold, :gain)
@@ -340,7 +338,7 @@ module DecisionTree
end end
def train(data = @data, attributes = @attributes, default = @default) 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) Ruleset.new(attributes, data, default, @type)
end end