huaxingao commented on a change in pull request #25947:
[SPARK-29269][PYTHON][ML] Pyspark ALSModel support getters/setters
URL: https://github.com/apache/spark/pull/25947#discussion_r329285366
##########
File path: python/pyspark/ml/recommendation.py
##########
@@ -258,133 +344,91 @@ def setImplicitPrefs(self, value):
"""
return self._set(implicitPrefs=value)
- @since("1.4.0")
- def getImplicitPrefs(self):
- """
- Gets the value of implicitPrefs or its default value.
- """
- return self.getOrDefault(self.implicitPrefs)
-
@since("1.4.0")
def setAlpha(self, value):
"""
Sets the value of :py:attr:`alpha`.
"""
return self._set(alpha=value)
- @since("1.4.0")
- def getAlpha(self):
- """
- Gets the value of alpha or its default value.
- """
- return self.getOrDefault(self.alpha)
-
@since("1.4.0")
def setUserCol(self, value):
"""
Sets the value of :py:attr:`userCol`.
"""
return self._set(userCol=value)
- @since("1.4.0")
- def getUserCol(self):
- """
- Gets the value of userCol or its default value.
- """
- return self.getOrDefault(self.userCol)
-
@since("1.4.0")
def setItemCol(self, value):
"""
Sets the value of :py:attr:`itemCol`.
"""
return self._set(itemCol=value)
- @since("1.4.0")
- def getItemCol(self):
- """
- Gets the value of itemCol or its default value.
- """
- return self.getOrDefault(self.itemCol)
-
@since("1.4.0")
def setRatingCol(self, value):
"""
Sets the value of :py:attr:`ratingCol`.
"""
return self._set(ratingCol=value)
- @since("1.4.0")
- def getRatingCol(self):
- """
- Gets the value of ratingCol or its default value.
- """
- return self.getOrDefault(self.ratingCol)
-
@since("1.4.0")
def setNonnegative(self, value):
"""
Sets the value of :py:attr:`nonnegative`.
"""
return self._set(nonnegative=value)
- @since("1.4.0")
- def getNonnegative(self):
- """
- Gets the value of nonnegative or its default value.
- """
- return self.getOrDefault(self.nonnegative)
-
@since("2.0.0")
def setIntermediateStorageLevel(self, value):
"""
Sets the value of :py:attr:`intermediateStorageLevel`.
"""
return self._set(intermediateStorageLevel=value)
- @since("2.0.0")
- def getIntermediateStorageLevel(self):
- """
- Gets the value of intermediateStorageLevel or its default value.
- """
- return self.getOrDefault(self.intermediateStorageLevel)
-
@since("2.0.0")
def setFinalStorageLevel(self, value):
"""
Sets the value of :py:attr:`finalStorageLevel`.
"""
return self._set(finalStorageLevel=value)
- @since("2.0.0")
- def getFinalStorageLevel(self):
- """
- Gets the value of finalStorageLevel or its default value.
- """
- return self.getOrDefault(self.finalStorageLevel)
-
@since("2.2.0")
def setColdStartStrategy(self, value):
"""
Sets the value of :py:attr:`coldStartStrategy`.
"""
return self._set(coldStartStrategy=value)
- @since("2.2.0")
- def getColdStartStrategy(self):
- """
- Gets the value of coldStartStrategy or its default value.
- """
- return self.getOrDefault(self.coldStartStrategy)
-
-class ALSModel(JavaModel, JavaMLWritable, JavaMLReadable):
+class ALSModel(JavaModel, _ALSModelParams, JavaMLWritable, JavaMLReadable):
"""
Model fitted by ALS.
.. versionadded:: 1.4.0
"""
+ @since("1.4.0")
+ def setUserCol(self, value):
Review comment:
Should be since(3.0.0). Thanks for catching that problem.
----------------------------------------------------------------
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]
