added comments

sileneer · Mar 7, 2023 · 1b2dc13 · 1b2dc13
1 parent 3395053
commit 1b2dc13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main_torch.ipynb b/main_torch.ipynb
@@ -2390,7 +2390,7 @@
     }
    ],
    "source": [
-    "for i in range(0, 4):\n",
+    "for i in range(0, 4): #plot both graphs\n",
     "    print(\"GRU Evaluation for Cell {}\".format(i + 5))\n",
     "    gru_outputs, targets, gru_MAE, gru_RMSE = evaluate(gru_model, test_x[i], test_y[i], label_sc)\n",
     "    plt.figure(figsize=(14, 10))\n",