diff options
Diffstat (limited to 'examples/src/main/python/ml/aft_survival_regression.py')
-rw-r--r-- | examples/src/main/python/ml/aft_survival_regression.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/src/main/python/ml/aft_survival_regression.py b/examples/src/main/python/ml/aft_survival_regression.py index 9879679829..060f0171ff 100644 --- a/examples/src/main/python/ml/aft_survival_regression.py +++ b/examples/src/main/python/ml/aft_survival_regression.py @@ -19,7 +19,7 @@ from __future__ import print_function # $example on$ from pyspark.ml.regression import AFTSurvivalRegression -from pyspark.mllib.linalg import Vectors +from pyspark.ml.linalg import Vectors # $example off$ from pyspark.sql import SparkSession |