fix train module typeerror

PengNi · Apr 4, 2022 · 7b18bb6 · 7b18bb6
1 parent 84b0fd6
commit 7b18bb6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/README.rst b/README.rst
@@ -4,6 +4,11 @@ deepsignal-plant
 
 Release
 -------
+0.1.6
+-----
+bug fixes (train module, )
+
+
 0.1.5
 -----
 make sure results of each read be written together in call_mods' output

diff --git a/deepsignal_plant/train.py b/deepsignal_plant/train.py
@@ -155,8 +155,8 @@ def train(args):
  # vprecs.append(i_precision)
  # vrecas.append(i_recall)
  vlosses.append(vloss.item())
- vlabels_total += vlabels
- vpredicted_total += vpredicted
+ vlabels_total += vlabels.tolist()
+ vpredicted_total += vpredicted.tolist()
 
  v_accuracy = metrics.accuracy_score(vlabels_total, vpredicted_total)
  v_precision = metrics.precision_score(vlabels_total, vpredicted_total)