diff options
| author | Xiangrui Meng <meng@databricks.com> | 2015-03-28 15:08:05 -0700 |
|---|---|---|
| committer | Joseph K. Bradley <joseph@databricks.com> | 2015-03-28 15:08:05 -0700 |
| commit | f75f633b21faaf911f04aeff847f25749b1ecd89 (patch) | |
| tree | 3cfa55e9e403283b4a10b2c5c56c4ca9e5289c5b /python/pyspark/mllib/recommendation.py | |
| parent | 99631438c0ec777d6a77974b148dbbd3e890260e (diff) | |
| download | spark-f75f633b21faaf911f04aeff847f25749b1ecd89.tar.gz spark-f75f633b21faaf911f04aeff847f25749b1ecd89.tar.bz2 spark-f75f633b21faaf911f04aeff847f25749b1ecd89.zip | |
[SPARK-6571][MLLIB] use wrapper in MatrixFactorizationModel.load
This fixes `predictAll` after load. jkbradley
Author: Xiangrui Meng <meng@databricks.com>
Closes #5243 from mengxr/SPARK-6571 and squashes the following commits:
82dcaa7 [Xiangrui Meng] use wrapper in MatrixFactorizationModel.load
Diffstat (limited to 'python/pyspark/mllib/recommendation.py')
| -rw-r--r-- | python/pyspark/mllib/recommendation.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/python/pyspark/mllib/recommendation.py b/python/pyspark/mllib/recommendation.py index 1a4527b12c..b094e50856 100644 --- a/python/pyspark/mllib/recommendation.py +++ b/python/pyspark/mllib/recommendation.py @@ -90,6 +90,8 @@ class MatrixFactorizationModel(JavaModelWrapper, JavaSaveable, JavaLoader): >>> sameModel = MatrixFactorizationModel.load(sc, path) >>> sameModel.predict(2,2) 0.43... + >>> sameModel.predictAll(testset).collect() + [Rating(... >>> try: ... os.removedirs(path) ... except OSError: @@ -111,6 +113,12 @@ class MatrixFactorizationModel(JavaModelWrapper, JavaSaveable, JavaLoader): def productFeatures(self): return self.call("getProductFeatures") + @classmethod + def load(cls, sc, path): + model = cls._load_java(sc, path) + wrapper = sc._jvm.MatrixFactorizationModelWrapper(model) + return MatrixFactorizationModel(wrapper) + class ALS(object): |