Cast threshold to string in gsub

igrigorik · Nov 23, 2015 · 823dd5e · 823dd5e
1 parent b8b315a
commit 823dd5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
@@ -190,7 +190,7 @@ def build_tree(tree = @tree)
         end
         label_text = "#{key} ''"
         if type(attr[0].attribute) == :continuous
-          label_text.gsub!("''", attr[0].threshold)
+          label_text.gsub!("''", attr[0].threshold.to_s)
         end
 
         [parent_text, child_text, label_text]