Fix

parcaster · Dec 2, 2023 · ac6ac9c · ac6ac9c
1 parent a6ccf5a
commit ac6ac9c
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/model/W&B_PPSG_LSTM.ipynb b/model/W&B_PPSG_LSTM.ipynb
@@ -418,21 +418,24 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "def inverse_transform(scaler, df, columns):\n",
-    "    for col in columns:\n",
-    "        df[col] = scaler.inverse_transform(df[col])\n",
-    "    return df\n",
+    "def inverse_scaler(scaler, df_output, df_target):\n",
+    "    output = scaler.inverse_transform(df_output)\n",
+    "    target = scaler.inverse_transform(df_target)\n",
+    "    \n",
+    "    df_output = pd.DataFrame(output, columns=parking_data_labels)\n",
+    "    df_target = pd.DataFrame(target, columns=parking_data_labels)\n",
+    "    \n",
+    "    return df_output, df_target\n",
     "\n",
     "def plot_test_prediction(scaler, outputs, targets):\n",
-    "    outputs = inverse_transform(scaler, pd.DataFrame(np.concatenate(outputs)), parking_data_labels)\n",
-    "    targets = inverse_transform(scaler, pd.DataFrame(np.concatenate(targets)), parking_data_labels)\n",
-    "\n",
     "    for i, (output, target) in enumerate(zip(outputs, targets)):\n",
     "        if i % 10 != 0:\n",
     "            continue\n",
     "\n",
     "        df_output = pd.DataFrame(output, columns=parking_data_labels)\n",
     "        df_target = pd.DataFrame(target, columns=parking_data_labels)\n",
+    "    \n",
+    "        df_output, df_target = inverse_scaler(scaler, df_output, df_target)\n",
     "\n",
     "        n_features = len(df_output.columns)\n",
     "\n",