Merge pull request #118 from vxld014/fix-mllib-adapter.py

vector.array does not work with sparse vectors. vector.toArray() work…
maxpumperla · Dec 1, 2018 · 5097175 · 5097175
2 parents e4b3b16 + b65a597
commit 5097175
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elephas/mllib/adapter.py b/elephas/mllib/adapter.py
@@ -23,7 +23,7 @@ def to_matrix(np_array):
 def from_vector(vector):
     """Convert MLlib Vector to numpy array
     """
-    return vector.array
+    return vector.toArray()
 
 
 def to_vector(np_array):