scala - How to reverse OneHotEncoderEstimator transformation in Spark

Question

So In Spark you can you OneHotEncoderEstimator to one-hot encode a column like this

import org.apache.spark.ml.feature.OneHotEncoderEstimator

val oneHotEncoder = new OneHotEncoderEstimator()
  .setInputCols(Array(...))
  .setOutputCols(...))
  .setDropLast(false)
oneHotEncoder.fit(...)

But how would you do the inverse operation? i.e. the equivalent of sklearn inverse_transform

Answer 1

等待大神解答