blob: 175146678faee659a323d9dcee8e32efcf844bcc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import sys
import timeit
print(timeit.timeit(
"""
PATH = "/home/munyoki/tmp/dataset/HLCPublish/10001/"
env = lmdb.open(PATH)
BLOB_HASH_DIGEST = 32
# def index_matrix(row_pointers):
with env.begin(write=False) as txn:
current_hash = txn.get(b"current")
matrix_hash = txn.get(current_hash + b":matrix")
row_pointers = txn.get(matrix_hash +
b":row-pointers")
nrows, = struct.unpack("<Q",
txn.get(matrix_hash + b":nrows"))
metadata = txn.get(matrix_hash + b":metadata")
sample_data = []
for i in range(0, (nrows-1)*32, 32):
sample_data.append(
json.loads(txn.get(row_pointers[i:i+32]).decode())
)
print(sample_data)
print(metadata.decode())
""",
setup="""
import struct
import array
import json
import lmdb
""",
number=int(sys.argv[1])
))
|
