From b53616026eac18d4e49bf5a60e077f7226afc951 Mon Sep 17 00:00:00 2001 From: Bago Amirbekian Date: Tue, 23 May 2017 18:01:45 -0700 Subject: [PATCH] BugFix - avoid passing float to ndarray.reshape in LogisticRegressionModel. --- python/pyspark/mllib/classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyspark/mllib/classification.py b/python/pyspark/mllib/classification.py index 9f53ed098202..e04eeb2b60d7 100644 --- a/python/pyspark/mllib/classification.py +++ b/python/pyspark/mllib/classification.py @@ -171,7 +171,7 @@ def __init__(self, weights, intercept, numFeatures, numClasses): self._dataWithBiasSize = None self._weightsMatrix = None else: - self._dataWithBiasSize = self._coeff.size / (self._numClasses - 1) + self._dataWithBiasSize = self._coeff.size // (self._numClasses - 1) self._weightsMatrix = self._coeff.toArray().reshape(self._numClasses - 1, self._dataWithBiasSize)