-
Notifications
You must be signed in to change notification settings - Fork 0
/
gen_test_complete.py
66 lines (53 loc) · 1.89 KB
/
gen_test_complete.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
"""
Code used to precompute splits and merges for baseline codes.
Only pickle data generated. Convert to numpy when needed.
"""
from __future__ import print_function
import numpy as np
import os, sys
from os import listdir
import Social_grouping as sg
sim = True
data_aug = True
a = sg.SocialGrouping(dataset = 'eth', flag = 0)
b = sg.SocialGrouping(dataset = 'eth', flag = 1)
c = sg.SocialGrouping(dataset = 'ucy', flag = 0)
d = sg.SocialGrouping(dataset = 'ucy', flag = 1)
e = sg.SocialGrouping(dataset = 'ucy', flag = 2)
#f = sg.SocialGrouping(dataset = 'ucy', flag = 3)
#g = sg.SocialGrouping(dataset = 'ucy', flag = 4)
num_split = 0
num_merge = 0
num_split += a.num_split
num_split += b.num_split
num_split += c.num_split
num_split += d.num_split
num_split += e.num_split
num_merge += a.num_merge
num_merge += b.num_merge
num_merge += c.num_merge
num_merge += d.num_merge
num_merge += e.num_merge
print('total number of split: ' + str(num_split))
print('total number of merge: ' + str(num_merge))
a.compile_positive_data(sim = sim, data_aug = data_aug)
a.compile_negative_data(int(a.num_split + a.num_merge)/2, sim = sim, data_aug = data_aug)
print('a')
b.compile_positive_data(sim = sim, data_aug = data_aug)
b.compile_negative_data(int(b.num_split + b.num_merge)/2, sim = sim, data_aug = data_aug)
print('b')
c.compile_positive_data(sim = sim, data_aug = data_aug)
c.compile_negative_data(int(c.num_split + c.num_merge)/2, sim = sim, data_aug = data_aug)
print('c')
d.compile_positive_data(sim = sim, data_aug = data_aug)
d.compile_negative_data(int(d.num_split + d.num_merge)/2, sim = sim, data_aug = data_aug)
print('d')
e.compile_positive_data(sim = sim, data_aug = data_aug)
e.compile_negative_data(int(e.num_split + e.num_merge)/2, sim = sim, data_aug = data_aug)
print('e')
#f.compile_positive_data()
#f.compile_negative_data()
#print('f')
#g.compile_positive_data()
#g.compile_negative_data()
#print('g')