org.deeplearning4j.ui.flow.beans

Class ModelInfo

java.lang.Object

org.deeplearning4j.ui.flow.beans.ModelInfo

All Implemented Interfaces:

Serializable

public class ModelInfo
extends Object
implements Serializable

This bean works as holder for unbounded list of layers. Each layer has it's own place in model's virtual coordinate space. For now, coordinate space is limited to 2 dimensions

Author:

raver119@gmail.com

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ModelInfo()

Method Summary

Modifier and Type	Method and Description
void	addLayer(LayerInfo layer) This method maps given layer into model coordinate space
LayerInfo	getLayerInfoByCoords(int x, int y) This method returns LayerInfo for specified grid coordinates
LayerInfo	getLayerInfoByName(String name) This method returns LayerInfo for specified layer name
Set<LayerInfo>	getLayersByType(String layerType)
Set<String>	getLayerTypes() This method returns all LayerTypes used in this model
int	size() This method returns the total number of nodes within described model

Methods inherited from class java.lang.Object

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

Constructor Detail

ModelInfo

public ModelInfo()

Method Detail

addLayer

public void addLayer(@NonNull
                     LayerInfo layer)

This method maps given layer into model coordinate space

Parameters:

layer -

getLayerInfoByName

public LayerInfo getLayerInfoByName(String name)

This method returns LayerInfo for specified layer name

Parameters:

name -

Returns:

getLayerInfoByCoords

public LayerInfo getLayerInfoByCoords(int x,
                                      int y)

This method returns LayerInfo for specified grid coordinates

Parameters:

x -

y -

Returns:

size

public int size()

This method returns the total number of nodes within described model

Returns:

number of elements

getLayerTypes

public Set<String> getLayerTypes()

This method returns all LayerTypes used in this model

Returns:

getLayersByType

public Set<LayerInfo> getLayersByType(String layerType)