def get_all_run_names(): runs = set() for subdir, dirs, files in os.walk( os.path.join(DataManager.get_data_folder(), "runs")): sub = subdir.split("runs")[1][1:].split("\\")[0].split("/")[0] if sub == "": continue runs.add(sub) return runs
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))