Fix

parcaster · Dec 2, 2023 · bd0b336 · bd0b336
1 parent f4c54b3
commit bd0b336
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model/W&B_PPSG_LSTM.ipynb b/model/W&B_PPSG_LSTM.ipynb
@@ -219,7 +219,7 @@
     "        X, y, input_dim, output_dim = load_features_labels(train_data_path)\n",
     "        X_train, X_val, y_train, y_val = split_train_val(X, y, config.train_val_ratio)\n",
     "        X_test, y_test, _, _ = load_features_labels(test_data_path)\n",
-    "        scaler = Scaler(StandardScaler)\n",
+    "        scaler = Scaler(StandardScaler())\n",
     "        X_train_scaled, X_val_scaled, X_test_scaled, y_train_scaled, y_val_scaled, y_test_scaled = scaler.scale(X_train,\n",
     "                                                                                                                X_val,\n",
     "                                                                                                                X_test,\n",
@@ -306,8 +306,8 @@
    "outputs": [],
    "source": [
     "def build_dataset(batch_size, X, y):\n",
-    "    features = torch.Tensor(X.values)\n",
-    "    targets = torch.Tensor(y.values)\n",
+    "    features = torch.Tensor(X)\n",
+    "    targets = torch.Tensor(y)\n",
     "\n",
     "    dataset = TensorDataset(features, targets)\n",
     "\n",