Commit dd81dbfe authored by Peter Eastman's avatar Peter Eastman
Browse files

Bug fix

parent 501a71bd
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -168,7 +168,7 @@
    "# We need a function which takes a 2d numpy array (samples, features) and returns predictions (samples,)\n",
    "def eval_model(my_model, transformers):\n",
    "    def eval_closure(x):\n",
    "        ds = dc.data.NumpyDataset(x, None, None, None)\n",
    "        ds = dc.data.NumpyDataset(x, None, None, None, n_tasks=n_tasks)\n",
    "        # The 0th task is NR-AR\n",
    "        predictions = model.predict(ds)[:,0]\n",
    "        return predictions\n",
@@ -36330,7 +36330,7 @@
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.5.6"
   "version": "3.6.9"
  }
 },
 "nbformat": 4,