50 from daal.data_management
import BlockDescriptor, CSRBlockDescriptor, CSRNumericTable, readOnly, readWrite
52 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
53 if utils_folder
not in sys.path:
54 sys.path.insert(0, utils_folder)
55 from utils
import printArray
58 if __name__ ==
"__main__":
60 print(
"Compressed spares rows (CSR) numeric table example\n")
68 values = np.array([1, -1, -3, -2, 5, 4, 6, 4, -4, 2, 7, 8, -5], dtype=np.float64)
69 colIndices = np.array([1, 2, 4, 1, 2, 3, 4, 5, 1, 3, 4, 2, 5], dtype=np.uint64)
70 rowOffsets = np.array([1, 4, 6, 9, 12, 14], dtype=np.uint64)
72 dataTable = CSRNumericTable(values, colIndices, rowOffsets, nFeatures, nObservations)
75 block = BlockDescriptor(ntype=np.float64)
76 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
77 print(str(block.getNumberOfRows()) +
" rows are read\n")
79 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
80 "Print 3 rows from CSR data array as dense double array:"
82 dataTable.releaseBlockOfRows(block)
85 csrBlock = CSRBlockDescriptor(ntpye=np.float32)
86 num_cols = csrBlock.getNumberOfColumns()
87 dataTable.getSparseBlock(firstReadRow, nRead, readWrite, csrBlock)
88 valuesBlock = csrBlock.getBlockValues()
89 nValuesInBlock = csrBlock.getDataSize()
90 printArray(valuesBlock, nValuesInBlock, 1, num_cols,
"Values in 3 rows from CSR data array:")
92 csrBlock.getBlockColumnIndices(), nValuesInBlock, 1, num_cols,
93 "Columns indices in 3 rows from CSR data array:", flt64=
False
96 csrBlock.getBlockRowIndices(), nRead + 1, 1, num_cols,
97 "Rows offsets in 3 rows from CSR data array:", flt64=
False
100 for i
in range(nValuesInBlock):
101 valuesBlock[i] = -(1.0 + i)
103 dataTable.releaseSparseBlock(csrBlock)
106 dataTable.getBlockOfRows(firstReadRow, nRead, readOnly, block)
107 print(str(block.getNumberOfRows()) +
" rows are read\n")
109 block.getArray(), nFeatures, block.getNumberOfRows(), block.getNumberOfColumns(),
110 "Print 3 rows from CSR data array as dense double array:"
112 dataTable.releaseBlockOfRows(block)