We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3a12a2d commit 9c329d8Copy full SHA for 9c329d8
python/pyspark/mllib/util.py
@@ -177,9 +177,9 @@ def appendBias(data):
177
"""
178
vec = _convert_to_vector(data)
179
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)
+ newIndices = np.append(vec.indices, len(vec))
+ newValues = np.append(vec.values, 1.0)
+ return SparseVector(len(vec) + 1, newIndices, newValues)
183
else:
184
return _convert_to_vector(np.append(vec.toArray(), 1.0))
185
0 commit comments