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