lib/eps/label_encoder.rb in eps-0.4.1 vs lib/eps/label_encoder.rb in eps-0.5.0

- old
+ new

@@ -34,10 +34,10 @@ end end end def inverse_transform(y) - inverse = Hash[@labels.map(&:reverse)] + inverse = @labels.map(&:reverse).to_h y.map do |yi| inverse[yi.to_i] end end end