Fix efficiency

maropu · Jun 23, 2015 · 9c329d8 · 9c329d8
1 parent 3a12a2d
commit 9c329d8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyspark/mllib/util.py b/python/pyspark/mllib/util.py
@@ -177,9 +177,9 @@ def appendBias(data):
         """
         vec = _convert_to_vector(data)
         if isinstance(vec, SparseVector):
-            entries = dict(zip(vec.indices, vec.values))
-            entries[len(vec)] = 1.0
-            return SparseVector(len(vec) + 1, entries)
+            newIndices = np.append(vec.indices, len(vec))
+            newValues = np.append(vec.values, 1.0)
+            return SparseVector(len(vec) + 1, newIndices, newValues)
         else:
             return _convert_to_vector(np.append(vec.toArray(), 1.0))