Example usage for org.deeplearning4j.nn.params DefaultParamInitializer WEIGHT_KEY

List of usage examples for org.deeplearning4j.nn.params DefaultParamInitializer WEIGHT_KEY

Introduction

In this page you can find the example usage for org.deeplearning4j.nn.params DefaultParamInitializer WEIGHT_KEY.

Prototype

String WEIGHT_KEY

To view the source code for org.deeplearning4j.nn.params DefaultParamInitializer WEIGHT_KEY.

Click Source Link

Usage

From source file:com.javafxpert.neuralnetviz.scenario.XorExample.java

License:Apache License

static void displayNetwork(MultiLayerNetwork mln) {
    System.out.println("multiLayerNetwork:");
    for (Layer layer : mln.getLayers()) {
        System.out.println("layer # " + layer.paramTable());
        INDArray w = layer.getParam(DefaultParamInitializer.WEIGHT_KEY);
        System.out.println("Weights: " + w);
        INDArray b = layer.getParam(DefaultParamInitializer.BIAS_KEY);
        System.out.println("Bias: " + b);
    }//w ww  .  j a v a2 s.  co  m
}