huaxingao commented on a change in pull request #26247: [SPARK-29566][ML]
Imputer should support single-column input/output
URL: https://github.com/apache/spark/pull/26247#discussion_r338851895
##########
File path: python/pyspark/ml/feature.py
##########
@@ -1308,6 +1357,20 @@ def setMissingValue(self, value):
"""
return self._set(missingValue=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)
+
Review comment:
I added the above setters explicitly because I need to add these later on
anyways.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
