- public class DeepLearningModel {
- public static void main(String[] args) {
- // 构建神经网络模型
- MultiLayerConfiguration config = new NeuralNetConfiguration.Builder()
- .seed(123)
- .activation(Activation.RELU)
- .weightInit(WeightInit.XAVIER)
- .list()
- .layer(new DenseLayer.Builder().nIn(784).nOut(250).build())
- .layer(new OutputLayer.Builder(LossFunctions.LossFunction.NEGATIVELOGLIKELIHOOD)
- .nIn(250).nOut(10).activation(Activation.SOFTMAX).build())
- .backprop(true).pretrain(false)
- .build();
-
- MultiLayerNetwork model =