-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathgenerateData.py
More file actions
executable file
·51 lines (42 loc) · 1.5 KB
/
Copy pathgenerateData.py
File metadata and controls
executable file
·51 lines (42 loc) · 1.5 KB
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
import random
if __name__ == '__main__':
filename = '/mnt/data/test/'
edgesFileName = filename + 'edges_raw.txt'
featsClassFileName = filename + 'featsClass_raw.txt'
edgesFile = open(edgesFileName, 'w+')
featsClassFile = open(featsClassFileName, 'w+')
# edgesFile.write('0\t1\n')
# edgesFile.write('1\t2\n')
# edgesFile.write('2\t3\n')
# edgesFile.write('3\t4\n')
# edgesFile.write('1\t0\n')
random.seed(1)
# insert nodes
for i in range(36):
str_node=str(i)
for j in range(34):
str_node+=('\t'+str(random.randint(-10,10)))
label=random.randint(0,1)
str_node+='\t'+str(label)+'\n'
featsClassFile.write(str_node)
edgeSet=set()
# for i in range(20):
# for j in range(1):
# rand_i=random.randint(0,19)
# while rand_i==i:
# rand_i=random.randint(0,19)
# if edgeSet.__contains__(str(i)+','+str(rand_i)):
# continue
# else:
# edgesFile.write(str(i)+'\t'+str(rand_i)+'\n')
# edgesFile.write(str(rand_i)+'\t'+str(i)+'\n')
# edgeSet.add(str(rand_i)+','+str(i))
for i in range(36):
for j in range(2):
nid=random.randint(0,35)
edgesFile.write(str(nid)+'\t'+str(i)+'\n')
edgesFile.write(str(i)+'\t'+str(nid)+'\n')
edgesFile.close()
# featsClassFile.write('0\t1\t0\ta\n')
# featsClassFile.write('1\t0\t1\ta\n')
featsClassFile.close()