}] 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)