Fixed forced conversion to stringjl

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