Cast threshold to string in gsub

This commit is contained in:
Danielius
2015-11-23 14:01:21 +02:00
parent b8b315a051
commit 823dd5e5fe

View File

@@ -190,7 +190,7 @@ module DecisionTree
end end
label_text = "#{key} ''" label_text = "#{key} ''"
if type(attr[0].attribute) == :continuous if type(attr[0].attribute) == :continuous
label_text.gsub!("''", attr[0].threshold) label_text.gsub!("''", attr[0].threshold.to_s)
end end
[parent_text, child_text, label_text] [parent_text, child_text, label_text]