26 from daal.data_management
import BlockDescriptor, CSRBlockDescriptor, CSRNumericTable, readOnly, readWrite
28 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
29 if utils_folder
not in sys.path:
30 sys.path.insert(0, utils_folder)
31 from utils
import printArray
34 if __name__ ==
"__main__":
36 print(
"Compressed spares rows (CSR) numeric table example\n")
44 values = np.array([1, -1, -3, -2, 5, 4, 6, 4, -4, 2, 7, 8, -5], dtype=np.float64)
45 colIndices = np.array([1, 2, 4, 1, 2, 3, 4, 5, 1, 3, 4, 2, 5], dtype=np.uint64)
46 rowOffsets = np.array([1, 4, 6, 9, 12, 14], dtype=np.uint64)
48 dataTable = CSRNumericTable(values, colIndices, rowOffsets, nFeatures, nObservations)
51 block = BlockDescriptor(ntype=np.float64)
52 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
53 print(str(block.getNumberOfRows()) +
" rows are read\n")
55 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
56 "Print 3 rows from CSR data array as dense double array:"
58 dataTable.releaseBlockOfRows(block)
61 csrBlock = CSRBlockDescriptor(ntpye=np.float32)
62 num_cols = csrBlock.getNumberOfColumns()
63 dataTable.getSparseBlock(firstReadRow, nRead, readWrite, csrBlock)
64 valuesBlock = csrBlock.getBlockValues()
65 nValuesInBlock = csrBlock.getDataSize()
66 printArray(valuesBlock, nValuesInBlock, 1, num_cols,
"Values in 3 rows from CSR data array:")
68 csrBlock.getBlockColumnIndices(), nValuesInBlock, 1, num_cols,
69 "Columns indices in 3 rows from CSR data array:", flt64=
False
72 csrBlock.getBlockRowIndices(), nRead + 1, 1, num_cols,
73 "Rows offsets in 3 rows from CSR data array:", flt64=
False
76 for i
in range(nValuesInBlock):
77 valuesBlock[i] = -(1.0 + i)
79 dataTable.releaseSparseBlock(csrBlock)
82 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
83 print(str(block.getNumberOfRows()) +
" rows are read\n")
85 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
86 "Print 3 rows from CSR data array as dense double array:"
88 dataTable.releaseBlockOfRows(block)