Projects/laydi
Projects
/
laydi
Archived
7
0
Fork 0
Code Issues Pull Requests Activity

Added convertion to sparse format in category dataset

This commit is contained in:
Arnar Flatberg 2008-01-07 13:13:27 +00:00
parent e6625c975f
commit bf29661af9
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
fluents/dataset.py
View File

@ -273,6 +273,19 @@ class CategoryDataset(Dataset):
def __init__(self, array, identifiers=None, name='C'): def __init__(self, array, identifiers=None, name='C'):
Dataset.__init__(self, array, identifiers=identifiers, name=name) Dataset.__init__(self, array, identifiers=identifiers, name=name)
def asspmatrix(self):
if isinstance(self._array, sparse.spmatrix):
return self._array
else:
arr = self.asarray()
return sparse.csr_matrix(arr.astype('i'))
def tospmatrix(self):
if isinstance(self._array, sparse.spmatrix):
self._array = self._array.tocsr()
else:
self._array = sparse.scr_matrix(self._array)
def as_dict_lists(self): def as_dict_lists(self):
"""Returns data as dict of identifiers along first dim. """Returns data as dict of identifiers along first dim.
@ -423,7 +436,7 @@ class Selection(dict):
def select(self, axis, labels): def select(self, axis, labels):
self[axis] = labels self[axis] = labels
def write_ftsv(fd, ds, decimals=7, sep='\t', fmt=None): def write_ftsv(fd, ds, decimals=7, sep='\t', fmt=None, sp_format=True):
"""Writes a dataset in fluents tab separated values (ftsv) form. """Writes a dataset in fluents tab separated values (ftsv) form.
@param fd: An open file descriptor to the output file. @param fd: An open file descriptor to the output file.
@ -466,6 +479,9 @@ def write_ftsv(fd, ds, decimals=7, sep='\t', fmt=None):
print >> fd, "# name: %s" % ds.get_name() print >> fd, "# name: %s" % ds.get_name()
# Write data # Write data
if hasattr(ds, "asspmatrix") and sp_format == True:
m = ds.asspmatrix()
else:
m = ds.asarray() m = ds.asarray()
if isinstance(m, sparse.spmatrix): if isinstance(m, sparse.spmatrix):
_write_sparse_elements(fd, m, fmt, sep) _write_sparse_elements(fd, m, fmt, sep)