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

Merge pull request #6 from superchris/master

Browse Source 
Fix graphing continuous attribute
This commit is contained in:
Ilya Grigorik
2012-10-28 11:20:34 -07:00
parent c70362de0c da25259a7a
commit a714dcecf7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/decisiontree/id3_tree.rb
View File

@@ -186,7 +186,7 @@ module DecisionTree
child = attr[1][key] child = attr[1][key]
child_text = "#{child}\n(#{child.to_s.clone.object_id})" child_text = "#{child}\n(#{child.to_s.clone.object_id})"
end end
label_text = "#{key} #{@type == :continuous ? attr[0].threshold : ""}" label_text = "#{key} #{type(attr[0].attribute) == :continuous ? attr[0].threshold : ""}"
[parent_text, child_text, label_text] [parent_text, child_text, label_text]
end end