diff options
Diffstat (limited to 'lib/prediction_data.rb')
-rw-r--r-- | lib/prediction_data.rb | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/prediction_data.rb b/lib/prediction_data.rb index 42da5fc..d387d24 100644 --- a/lib/prediction_data.rb +++ b/lib/prediction_data.rb @@ -270,7 +270,7 @@ module Lib def self.classification_vals(dataset, compound, feature, accept_values) v_indices = [] values(dataset, compound, feature).each do |v| - i = accept_values.index(v.to_s) + i = accept_values.index(v) raise "illegal class_value of prediction (value is '"+v.to_s+"'), accept values are "+ accept_values.inspect unless v==nil or i!=nil v_indices << i @@ -294,4 +294,4 @@ module Lib v_mod end end -end
\ No newline at end of file +end |