-
Notifications
You must be signed in to change notification settings - Fork 7
/
h1_evaluation.py
41 lines (34 loc) · 1.05 KB
/
h1_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
import helpers
import kapa
import sys
import csv
digit_names = ['0', '1', '2', '3', '4', '6', 'period', '9']
#digit_names = ['left', 'right']
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
affinity_graph = []
for i in range(30):
affinity_graph += kapa.kapa(
antigens = [helpers.create_antigen(name) for name in digit_names],
antibodies = [kapa.Antibody() for _ in range(10)],
generations = 60,
num_clone_antibodies = 5,
num_kill_antibodies = 0,
clone_multiplier = 10
)
for affinity_rating in affinity_graph:
affinity_rating['key'] = digit_names[affinity_rating['key']]
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()