lib/eps/evaluators/lightgbm.rb in eps-0.3.2 vs lib/eps/evaluators/lightgbm.rb in eps-0.3.3

@@ -14,11 +14,11 @@
       def predict(data)
         rows = data.map(&:to_h)
 
         # sparse matrix
         @text_features.each do |k, v|
-          encoder = TextEncoder.new(v)
+          encoder = TextEncoder.new(**v)
 
           values = data.columns.delete(k)
           counts = encoder.transform(values)
 
           encoder.vocabulary.each do |word|
@@ -79,10 +79,10 @@
             # missingValueStrategy="none"
             false
           else
             case node.operator
             when "equal"
-              v == node.value
+              v.to_s == node.value
             when "in"
               node.value.include?(v)
             when "greaterThan"
               v > node.value
             when "lessOrEqual"