From 823dd5e5fe984d31839b4a48fe6c392f68c3a15e Mon Sep 17 00:00:00 2001 From: Danielius Date: Mon, 23 Nov 2015 14:01:21 +0200 Subject: [PATCH] Cast threshold to string in gsub --- lib/decisiontree/id3_tree.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb index 4e4c340..b0702c8 100755 --- a/lib/decisiontree/id3_tree.rb +++ b/lib/decisiontree/id3_tree.rb @@ -190,7 +190,7 @@ module DecisionTree 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]