}]
with open('best_individual.json') as best_agent_weights:
weights = json.load(best_agent_weights)
num_inputs = 6
num_outputs = 3
a = Ann(num_inputs, num_outputs)
a.weights = weights
edges = []
i = 0
for output_index in range(num_outputs):
for input_index in range(num_inputs + 1):
idx = a.convert_2d_to_1d(input_index, output_index)
weight = a.weights[idx]
edges.append({
'id': 'e{}'.format(i),
'source': 'i{}'.format(input_index),
'target': 'o{}'.format(output_index),
'weight': weight
})
i += 1
data = {'nodes': nodes, 'edges': edges}
with open('best_individual_nn.json', 'w') as outfile:
json.dump(data, outfile)
