# import pdb # import pandas as pd # import csv from data_formatter import DataFormatter ## Load Data ## path_to_dataset = "../../Data/train_100_events/" event_path = "event000001052" model_name = "identity.keras" hits, cells, particles, truth = load_event(path_to_dataset + event_path) # Get the sorted tracks formatter = DataFormatter() true_tracks, hit_tracks = formatter.getSortedTracks(particles, truth, hits) ## Load Predicted Seeds ## seed_file = open("SeedCandidates.txt", "r") our_tracks = [] seed_hits = [] np_hits = np.asarray(hits) for seed_id in seed_file: seed_id = int(float(seed_id.strip())) seed_hit = np_hits[np_hits[:, 0] == seed_id][0] our_tracks.append([int(seed_hit[0])]) seed_hits.append(seed_hit) print("\nStarting with " + str(len(seed_hits)) + " seed hits") ## Evaluate Predicted Seeds ##