aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/__init__.py
diff options
context:
space:
mode:
authorTor Myklebust <tmyklebu@gmail.com>2013-12-19 22:45:16 -0500
committerTor Myklebust <tmyklebu@gmail.com>2013-12-19 22:45:16 -0500
commit2328bdd00f701ca3b1bc7fdf8b2968fafc58fd11 (patch)
treed17922b3b78cdc7eebf121b653fdf5010dc279e0 /python/pyspark/__init__.py
parentded67ee90c2c0b22d67e623156a3f6cce8573abd (diff)
downloadspark-2328bdd00f701ca3b1bc7fdf8b2968fafc58fd11.tar.gz
spark-2328bdd00f701ca3b1bc7fdf8b2968fafc58fd11.tar.bz2
spark-2328bdd00f701ca3b1bc7fdf8b2968fafc58fd11.zip
Python side of python bindings for linear, Lasso, and ridge regression
Diffstat (limited to 'python/pyspark/__init__.py')
-rw-r--r--python/pyspark/__init__.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/python/pyspark/__init__.py b/python/pyspark/__init__.py
index 9f71db397d..7c8f9148d5 100644
--- a/python/pyspark/__init__.py
+++ b/python/pyspark/__init__.py
@@ -42,7 +42,9 @@ from pyspark.context import SparkContext
from pyspark.rdd import RDD
from pyspark.files import SparkFiles
from pyspark.storagelevel import StorageLevel
-from pyspark.mllib import LinearRegressionModel
+from pyspark.mllib import LinearRegressionModel, LassoModel, \
+ RidgeRegressionModel
-__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", "LinearRegressionModel"];
+__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", \
+ "LinearRegressionModel", "LassoModel", "RidgeRegressionModel"];