def main():
nodes = setup_nodes()
transitions = setup_initial_transitions(nodes)
learned_transitions = []
walked_transitions = []
# Start at Arad
current_node = Node.get_node_by_value("Lugoj", nodes)
# Steps
while (not (objective_function(current_node, nodes))):
new_transition = step_function(current_node, learned_transitions,
transitions)
walked_transitions.append(new_transition)
current_node = new_transition.to
current_node.already_walked = new_transition.weight - current_node.heuristic
print("\n")
Transition.print_transitions_array(walked_transitions)
def objective_function(current_node, nodes):
# End at Bucharest
objective = Node.get_node_by_value("Bucharest", nodes)
return current_node == objective
