Skip to content

Commit 9c329d8

Browse files
committed
Fix efficiency
1 parent 3a12a2d commit 9c329d8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/pyspark/mllib/util.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ def appendBias(data):
177177
"""
178178
vec = _convert_to_vector(data)
179179
if isinstance(vec, SparseVector):
180-
entries = dict(zip(vec.indices, vec.values))
181-
entries[len(vec)] = 1.0
182-
return SparseVector(len(vec) + 1, entries)
180+
newIndices = np.append(vec.indices, len(vec))
181+
newValues = np.append(vec.values, 1.0)
182+
return SparseVector(len(vec) + 1, newIndices, newValues)
183183
else:
184184
return _convert_to_vector(np.append(vec.toArray(), 1.0))
185185

0 commit comments

Comments
 (0)