Merge pull request #1 from Adolfhill/Adolfhill-patch-1

Adolfhill · web-flow · commit 0e97cacd51df · 2024-01-15T22:24:50.000+08:00
Update test.py
diff --git a/test.py b/test.py
@@ -24,7 +24,7 @@
     reader = csv.reader(f)
     next(reader)
     for row in reader:
-        names[int(row[0])] = row[5].split(";")[0]
+        names[int(row[0]) - 1] = row[5].split(";")[0]
 
 
 def visualize_result(data, pred, cfg):
