File tree 1 file changed +1
-11
lines changed
dl4j-examples/src/main/kotlin/org/deeplearning4j/examples/feedforward/mnist
1 file changed +1
-11
lines changed Original file line number Diff line number Diff line change 16
16
17
17
package org.deeplearning4j.examples.feedforward.mnist
18
18
19
-
20
- import org.nd4j.linalg.activations.Activation
21
- import org.nd4j.linalg.dataset.api.iterator.DataSetIterator
22
19
import org.deeplearning4j.datasets.iterator.impl.MnistDataSetIterator
23
- import org.deeplearning4j.eval.Evaluation
24
- import org.deeplearning4j.nn.api.OptimizationAlgorithm
25
- import org.deeplearning4j.nn.conf.MultiLayerConfiguration
26
20
import org.deeplearning4j.nn.conf.NeuralNetConfiguration
27
- import org.deeplearning4j.nn.conf.Updater
28
21
import org.deeplearning4j.nn.conf.layers.DenseLayer
29
22
import org.deeplearning4j.nn.conf.layers.OutputLayer
30
23
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork
31
24
import org.deeplearning4j.nn.weights.WeightInit
32
25
import org.deeplearning4j.optimize.listeners.ScoreIterationListener
33
- import org.nd4j.linalg.api.ndarray.INDArray
34
- import org.nd4j.linalg.dataset.DataSet
26
+ import org.nd4j.linalg.activations.Activation
35
27
import org.nd4j.linalg.learning.config.Nesterovs
36
28
import org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction
37
- import org.slf4j.Logger
38
29
import org.slf4j.LoggerFactory
39
30
40
-
41
31
/* * A slightly more involved multilayered (MLP) applied to digit classification for the MNIST dataset (http://yann.lecun.com/exdb/mnist/).
42
32
43
33
* This example uses two input layers and one hidden layer.
You can’t perform that action at this time.
0 commit comments