-
Notifications
You must be signed in to change notification settings - Fork 7
/
h2_evaluation.py
48 lines (43 loc) · 1.25 KB
/
h2_evaluation.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
import helpers
import kapa
import sys
import csv
def print_shape(shape):
x = -1
for i in range(12):
row = ''
for j in range(10):
x += 1
if shape[x]:
row += 'X'
else:
row += ' '
print row
results = []
for i in range(1, 11):
for _ in range(30):
result = kapa.kapa(
antigens = [kapa.Antibody() for _ in range(i)],
antibodies = [kapa.Antibody() for _ in range(10)],
generations = 100,
num_clone_antibodies = 5,
num_kill_antibodies = 0,
clone_multiplier = 10
)
print result
results.append({
'antigens' : i,
'generations' : result
})
with open('output.csv', 'wb') as output_csv:
dict_writer = csv.DictWriter(output_csv, ['antigens', 'generations'])
dict_writer.writeheader()
dict_writer.writerows(results)
sys.exit()
with open('output.csv', 'wb') as output_csv:
dict_writer = csv.DictWriter(output_csv, ['generation', 'key', 'affinity'])
dict_writer.writeheader()
dict_writer.writerows(affinity_graph)
sys.exit()
for antibody in best_antibodies:
print_shape(antibody.shape)