fix key

dsethz · Nov 9, 2024 · 7b35ca2 · 7b35ca2
1 parent 13920c4
commit 7b35ca2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/nuclai/models/evaluate_lr_rf.py b/src/nuclai/models/evaluate_lr_rf.py
@@ -66,7 +66,7 @@ def main():
         data = json.load(f)
 
     # compile training data
-    tmp = np.load(data[0]["path"])
+    tmp = np.load(data["0"]["path"])
     data_test = np.zeros((len(data), tmp.shape[1]))
     labels_test = np.zeros(len(data))
     ids = []

diff --git a/src/nuclai/models/train_lr_rf.py b/src/nuclai/models/train_lr_rf.py
@@ -59,7 +59,7 @@ def main():
         data = json.load(f)
 
     # compile training data
-    tmp = np.load(data[0]["path"])
+    tmp = np.load(data["0"]["path"])
     data_train = np.zeros((len(data), tmp.shape[1]))
     labels_train = np.zeros(len(data))