def plot_best_graphs(run_name, top_num=1, view=True):
# try:
generation = DataManager.load_generation_state(run_name)
# except Exception as e:
# print(e)
# return
best_graphs = generation.pareto_population.get_highest_accuracy(
num=top_num, check_set=generation.pareto_population.best_members)
if top_num == 1:
best_graphs = [best_graphs]
for graph in best_graphs:
sample, _ = DataLoader.sample_data(Config.get_device(),
dataset=graph.dataset)
model = create_nn(graph, sample, feature_multiplier=1)
file_name = run_name + "_best" + repr(best_graphs.index(graph))
graph.plot_tree_with_graphvis(
view=view,
file=os.path.join(DataManager.get_data_folder(), "runs", run_name,
"Graphs", file_name))
def continue_evolution_from_save_state(run_name):
generation = DataManager.load_generation_state(run_name)
generation.update_rank_function()
evolve_from_generation(generation)
