diff --git a/notebooks/Convolutional layer tests.ipynb b/notebooks/Convolutional layer tests.ipynb index 5870d46..8c085cc 100644 --- a/notebooks/Convolutional layer tests.ipynb +++ b/notebooks/Convolutional layer tests.ipynb @@ -15,10 +15,13 @@ "cell_type": "code", "execution_count": null, "metadata": { - "collapsed": true + "collapsed": false }, "outputs": [], "source": [ + "import mlp.layers as layers\n", + "import mlp.initialisers as init\n", + "\n", "class ConvolutionalLayer(layers.LayerWithParameters):\n", " \"\"\"Layer implementing a 2D convolution-based transformation of its inputs.\n", "\n", @@ -187,6 +190,8 @@ }, "outputs": [], "source": [ + "import numpy as np\n", + "\n", "def test_conv_layer_fprop(layer_class, do_cross_correlation=False):\n", " \"\"\"Tests `fprop` method of a convolutional layer.\n", " \n",