-
Notifications
You must be signed in to change notification settings - Fork 1
/
generate_manifests.py
107 lines (98 loc) · 1.8 KB
/
generate_manifests.py
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import yaml
import sys
# A proper manifest would have flow filters instead of thr
manifest = {
'sketches': [
{
'rows': 4,
'cols': 32768,
'thr': 1,
'frac': 1
},
{
'rows': 4,
'cols': 32768,
'thr': 1,
'frac': 1
}
],
'total_thr': 2
}
hash_bench = [
{
'sketches': [
{
'rows': r,
'cols': 4,
'thr': 1,
'frac': 1
}
],
'total_thr': 1
}
for r in range(1, 13)
]
mem_bench = [
{
'sketches': [
{
'rows': r,
'cols': 2**logc,
'thr': 1,
'frac': 1
}
],
'total_thr': 1
}
for r in [10, 12]
for logc in range(23)
]
ground_truth = [
{
'sketches': [
{
'rows': r,
'cols': 4**logc,
'thr': 1,
'frac': 1
}
],
'total_thr': 1
}
for r in range(2, 10, 2)
for logc in range(2, 12)
]
ground_truth_small = [
{
'sketches': [
{
'rows': r,
'cols': 4**logc,
'thr': 1,
'frac': 1,
'levels': l
}
],
'total_thr': 1
}
for r in [3, 6, 9]
for logc in range(2, 10, 2)
for l in [4, 8, 16]
]
testing = [
{
'sketches': [
{
'rows': 8,
'cols': 4**logc,
'thr': 1,
'frac': 1
}
],
'total_thr': 1
}
for logc in range(10, 12)
]
# double_sketches = {
# }
yaml.dump(ground_truth_small, sys.stdout)