Error al importar OneHotEncoderEstimator

Problema

Ha migrado un cuaderno de Databricks Runtime 6.4 para Machine Learning o inferior a Databricks Runtime 7.3 para Machine Learning o superior.

Está intentando importar y OneHotEncoderEstimator recibe un error de importación.

ImportError: cannot import name 'OneHotEncoderEstimator' from 'pyspark.ml.feature' (/databricks/spark/python/pyspark/ml/feature.py)

Causa

OneHotEncoderEstimator se ha cambiado el nombre OneHotEncoder a en Apache Spark 3.0.

Solución

Debe reemplazar las OneHotEncoderEstimator referencias del cuaderno por OneHotEncoder .

Por ejemplo, el código de ejemplo siguiente devuelve un error de importación en Databricks Runtime 7.3 para Machine Learning o superior:

from pyspark.ml.feature import OneHotEncoderEstimator

El código de ejemplo siguiente funciona correctamente en Databricks Runtime 7.3 para Machine Learning o superior:

from pyspark.ml.feature import OneHotEncoder