From d35f7348856829fa072cefbcd3c702b7ac739a98 Mon Sep 17 00:00:00 2001
From: Illia Oleksiienko <io@ece.au.dk>
Date: Sat, 11 Feb 2023 16:31:05 +0000
Subject: [PATCH] Update plot generation
---
run/create_plots.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/run/create_plots.py b/run/create_plots.py
index fda5204fdf..c2d7d544e7 100644
--- a/run/create_plots.py
+++ b/run/create_plots.py
@@ -48,7 +48,7 @@ def read_data(file, frame):
network = "_".join(model_name.split("_")[1:])
method = words[1] if len(words) > 2 else words[0]
- if "vnnclass1" in model_name:
+ if "vnnclass1" or "vnnclass3" in model_name:
method=words[-3]
if model_name in ["tanet_car_2", "tanet_car_3"]:
--
GitLab