class DNN::Models::LayersList

This class is used to hold multiple layers in an array.

Public Class Methods

from_hash_list(hash_list) click to toggle source
# File lib/dnn/core/models.rb, line 6
def self.from_hash_list(hash_list)
  layers_list = new
  hash_list.each do |hash|
    obj_class = DNN.const_get(hash[:class])
    obj = obj_class.allocate
    if obj.is_a?(Chain)
      obj = obj_class.new
      obj.load_hash(hash)
    else
      obj = Layers::Layer.from_hash(hash)
    end
    layers_list << obj
  end
  layers_list
end

Public Instance Methods

layers() click to toggle source

Get the all layers. @return [Array] All layers array.

# File lib/dnn/core/models.rb, line 28
def layers
  layers_array = []
  each do |layer|
    if layer.is_a?(Layers::Layer)
      layers_array << layer
    elsif layer.is_a?(Chain) || layer.is_a?(LayersList)
      layers_array.concat(layer.layers)
    end
  end
  layers_array
end
to_hash_list() click to toggle source
# File lib/dnn/core/models.rb, line 22
def to_hash_list
  map(&:to_hash)
end