List of usage examples for org.deeplearning4j.nn.graph ComputationGraph ComputationGraph
public ComputationGraph(ComputationGraphConfiguration configuration)
From source file:gr.aueb.cs.nlp.wordtagger.classifier.DeepNetClassifier.java
License:Open Source License
public DeepNetClassifier(Model model, boolean useHistogramListener, ComputationGraphConfiguration conf) { this.model = model; this.net = new ComputationGraph(conf); if (useHistogramListener) net.setListeners(Arrays.asList(new HistogramIterationListener(100))); this.net.init(); }