From 1adc184edcb95d6241007404d0fe1bdff0e31f76 Mon Sep 17 00:00:00 2001 From: zero323 Date: Mon, 13 Jan 2020 23:16:55 +0100 Subject: [PATCH] Add setInputCol and setOutputCol to ImputerModel --- python/pyspark/ml/feature.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/pyspark/ml/feature.py b/python/pyspark/ml/feature.py index 839386b3a4c16..32e90496de300 100755 --- a/python/pyspark/ml/feature.py +++ b/python/pyspark/ml/feature.py @@ -1714,6 +1714,20 @@ def setOutputCols(self, value): """ return self._set(outputCols=value) + @since("3.0.0") + def setInputCol(self, value): + """ + Sets the value of :py:attr:`inputCol`. + """ + return self._set(inputCol=value) + + @since("3.0.0") + def setOutputCol(self, value): + """ + Sets the value of :py:attr:`outputCol`. + """ + return self._set(outputCol=value) + @property @since("2.2.0") def surrogateDF(self):