improve solver

Corentin-Allaire · Feb 17, 2025 · 4ce1e53 · 4ce1e53
1 parent f9638f8
commit 4ce1e53
Show file tree

Hide file tree

Showing 3 changed files with 209 additions and 89 deletions.
diff --git a/Examples/Scripts/Python/MLAmbiguityResolution/match_good_track-seed.py b/Examples/Scripts/Python/MLAmbiguityResolution/match_good_track-seed.py
@@ -22,7 +22,10 @@ def matchGood(seed_files: list[str], ckf_files: list[str]):
         data_seed = pd.read_csv(f_seed)
         # Add a good seed column to the seed dataset
         data_seed["goodSeed"] = data_seed["seed_id"].isin(goodSeed)
-
+        # Add a rank column 
+        data_seed["rank"] = -1
+        # if the seed match a track copy the track rank
+        data_seed.loc[data_seed["seed_id"].isin(data_track["seed_id"]), "rank"] = data_track["rank"]
         data_seed.loc[
             data_seed["good/duplicate/fake"] == "good", "good/duplicate/fake"
         ] = "duplicate"