Skip to content
Snippets Groups Projects
Commit 29d34c7e authored by Yannick DAYER's avatar Yannick DAYER
Browse files

Merge branch 'finalize-scores-tweak' into 'master'

finalize-scores tweak

See merge request !103
parents 31ef6493 98d0f748
No related branches found
No related tags found
1 merge request!103finalize-scores tweak
Pipeline #63243 passed
...@@ -63,13 +63,13 @@ def finalize_scores(scores, method, backup, verbose): ...@@ -63,13 +63,13 @@ def finalize_scores(scores, method, backup, verbose):
df = pd.read_csv(path) df = pd.read_csv(path)
# average the scores of each frame # average the scores of each frame
df["score"] = df.groupby("test_label")["score"].transform(mean) df["score"] = df.groupby("video_key")["score"].transform(mean)
# remove frame_id column if it exists # remove frame_id column if it exists
if "frame_id" in df.columns: if "frame_id" in df.columns:
df.drop("frame_id", axis=1, inplace=True) df.drop("frame_id", axis=1, inplace=True)
# make rows unique based on test_label # make rows unique based on test_label
df.drop_duplicates(subset=["test_label"], inplace=True) df.drop_duplicates(subset=["video_key"], inplace=True)
df.to_csv(path, index=False) df.to_csv(path, index=False)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment