Skip to content
Snippets Groups Projects
Commit 73a99eec authored by bbauvin's avatar bbauvin
Browse files

Trying to fix segmentation fault

parent 9604c3f5
No related branches found
No related tags found
No related merge requests found
...@@ -35,14 +35,14 @@ def getValue(DATASET): ...@@ -35,14 +35,14 @@ def getValue(DATASET):
def extractSubset(matrix, usedIndices): def extractSubset(matrix, usedIndices):
if sparse.issparse(matrix): if sparse.issparse(matrix):
newIndptr = np.zeros(len(usedIndices)+1, dtype=np.int16) newIndptr = np.zeros(len(usedIndices)+1, dtype=int)
oldindptr = matrix.indptr oldindptr = matrix.indptr
print oldindptr print oldindptr
for exampleIndexIndex, exampleIndex in enumerate(usedIndices): for exampleIndexIndex, exampleIndex in enumerate(usedIndices):
newIndptr[exampleIndexIndex+1] = newIndptr[exampleIndexIndex]+(oldindptr[exampleIndex+1]-oldindptr[exampleIndex]) newIndptr[exampleIndexIndex+1] = newIndptr[exampleIndexIndex]+(oldindptr[exampleIndex+1]-oldindptr[exampleIndex])
print newIndptr print newIndptr
newData = np.ones(newIndptr[-1], dtype=bool) newData = np.ones(newIndptr[-1], dtype=bool)
newIndices = np.zeros(newIndptr[-1], dtype=np.int32) newIndices = np.zeros(newIndptr[-1], dtype=int)
oldIndices = matrix.indices oldIndices = matrix.indices
for exampleIndexIndex, exampleIndex in enumerate(usedIndices): for exampleIndexIndex, exampleIndex in enumerate(usedIndices):
newIndices[newIndptr[exampleIndexIndex]:newIndptr[exampleIndexIndex+1]] = oldIndices[oldindptr[exampleIndex]: oldindptr[exampleIndex+1]] newIndices[newIndptr[exampleIndexIndex]:newIndptr[exampleIndexIndex+1]] = oldIndices[oldindptr[exampleIndex]: oldindptr[exampleIndex+1]]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment