diff --git a/src/ptbench/engine/road_calculator.py b/src/ptbench/engine/road_calculator.py index f2e2a82e112e8f008ef0408de9a4e26a9f9d2f93..d4ebd329cd9ced34239b0a1d9a670dbba599dfd6 100644 --- a/src/ptbench/engine/road_calculator.py +++ b/src/ptbench/engine/road_calculator.py @@ -118,7 +118,7 @@ def process_target_class( # Write metrics to csv file csv_writer.writerow( [ - names[0].split("/")[1], + names[0], MoRF_score, LeRF_score, combined_score,