49 from daal.data_management
import BlockDescriptor, CSRBlockDescriptor, CSRNumericTable, readOnly, readWrite
51 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
52 if utils_folder
not in sys.path:
53 sys.path.insert(0, utils_folder)
54 from utils
import printArray
57 if __name__ ==
"__main__":
59 print(
"Compressed spares rows (CSR) numeric table example\n")
67 values = np.array([1, -1, -3, -2, 5, 4, 6, 4, -4, 2, 7, 8, -5], dtype=np.float64)
68 colIndices = np.array([1, 2, 4, 1, 2, 3, 4, 5, 1, 3, 4, 2, 5], dtype=np.uint64)
69 rowOffsets = np.array([1, 4, 6, 9, 12, 14], dtype=np.uint64)
71 dataTable = CSRNumericTable(values, colIndices, rowOffsets, nFeatures, nObservations)
74 block = BlockDescriptor(ntype=np.float64)
75 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
76 print(str(block.getNumberOfRows()) +
" rows are read\n")
78 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
79 "Print 3 rows from CSR data array as dense double array:"
81 dataTable.releaseBlockOfRows(block)
84 csrBlock = CSRBlockDescriptor(ntpye=np.float32)
85 num_cols = csrBlock.getNumberOfColumns()
86 dataTable.getSparseBlock(firstReadRow, nRead, readWrite, csrBlock)
87 valuesBlock = csrBlock.getBlockValues()
88 nValuesInBlock = csrBlock.getDataSize()
89 printArray(valuesBlock, nValuesInBlock, 1, num_cols,
"Values in 3 rows from CSR data array:")
91 csrBlock.getBlockColumnIndices(), nValuesInBlock, 1, num_cols,
92 "Columns indices in 3 rows from CSR data array:", flt64=
False
95 csrBlock.getBlockRowIndices(), nRead + 1, 1, num_cols,
96 "Rows offsets in 3 rows from CSR data array:", flt64=
False
99 for i
in range(nValuesInBlock):
100 valuesBlock[i] = -(1.0 + i)
102 dataTable.releaseSparseBlock(csrBlock)
105 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
106 print(str(block.getNumberOfRows()) +
" rows are read\n")
108 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
109 "Print 3 rows from CSR data array as dense double array:"
111 dataTable.releaseBlockOfRows(block)