org.deeplearning4j.ui.module.train

Class TrainModuleUtils

java.lang.Object

org.deeplearning4j.ui.module.train.TrainModuleUtils

public class TrainModuleUtils
extends Object

Utility methods for TrainModule

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TrainModuleUtils.GraphInfo

Constructor Summary

Constructors

Constructor and Description
TrainModuleUtils()

Method Summary

Modifier and Type	Method and Description
static TrainModuleUtils.GraphInfo	buildGraphInfo(org.deeplearning4j.nn.conf.ComputationGraphConfiguration config)
static TrainModuleUtils.GraphInfo	buildGraphInfo(org.deeplearning4j.nn.conf.MultiLayerConfiguration config)
static TrainModuleUtils.GraphInfo	buildGraphInfo(org.deeplearning4j.nn.conf.NeuralNetConfiguration config)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TrainModuleUtils

public TrainModuleUtils()

Method Detail

buildGraphInfo

public static TrainModuleUtils.GraphInfo buildGraphInfo(org.deeplearning4j.nn.conf.MultiLayerConfiguration config)

buildGraphInfo

public static TrainModuleUtils.GraphInfo buildGraphInfo(org.deeplearning4j.nn.conf.ComputationGraphConfiguration config)

buildGraphInfo

public static TrainModuleUtils.GraphInfo buildGraphInfo(org.deeplearning4j.nn.conf.NeuralNetConfiguration config)