導入OneHotEncoderEstimator時出錯

當試圖導入OneHotEncoderEstimator時,會收到一條錯誤消息。

寫的shyamprasad.miryala

最新發布日期:2022年5月16日

問題

你已經將筆記本從Databricks Runtime 6.4(機器學習)或更低版本遷移到Databricks Runtime 7.3(機器學習)或更高版本。

您正在嚐試導入OneHotEncoderEstimator你會得到一個導入錯誤。

ImportError:無法從'pyspark.ml '中導入名稱'OneHotEncoderEstimator'。功能”(/磚/火花/ python / pyspark / ml / feature.py)

導致

OneHotEncoderEstimator被重命名OneHotEncoderApache Spark 3.0。

解決方案

你必須更換OneHotEncoderEstimator參考資料在你的筆記本OneHotEncoder

例如,以下示例代碼在用於機器學習或更高版本的Databricks Runtime 7.3中返回一個導入錯誤:

%python from pyspark.ml.feature import OneHotEncoderEstimator

以下示例代碼在Databricks Runtime 7.3(適用於機器學習或更高版本)中正確運行:

從pyspark.ml.feature導入OneHotEncoder