diff options
Diffstat (limited to 'python/pyspark/__init__.py')
-rw-r--r-- | python/pyspark/__init__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/pyspark/__init__.py b/python/pyspark/__init__.py index 8b5bb79a18..3d73d95909 100644 --- a/python/pyspark/__init__.py +++ b/python/pyspark/__init__.py @@ -43,9 +43,10 @@ from pyspark.rdd import RDD from pyspark.files import SparkFiles from pyspark.storagelevel import StorageLevel from pyspark.mllib import LinearRegressionModel, LassoModel, \ - RidgeRegressionModel, LogisticRegressionModel, SVMModel, KMeansModel + RidgeRegressionModel, LogisticRegressionModel, SVMModel, KMeansModel, \ + ALSModel __all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", "LinearRegressionModel", "LassoModel", "RidgeRegressionModel", - "LogisticRegressionModel", "SVMModel", "KMeansModel"]; + "LogisticRegressionModel", "SVMModel", "KMeansModel", "ALSModel"]; |