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

Broke graphing tree with continuous attributes

Browse Source
This commit is contained in:
Chris Nelson
2012-10-28 12:41:26 -04:00
parent d1dce9be91
commit da25259a7a
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_text = "#{child}\n(#{child.to_s.clone.object_id})"
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]
end