class TensorFlow::Keras::Models::Sequential
Public Class Methods
new(layers = [])
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 5 def initialize(layers = []) @layers = [] layers.each do |layer| add(layer) end end
Public Instance Methods
add(layer)
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 13 def add(layer) @layers << layer end
compile(optimizer: nil, loss: nil, metrics: nil)
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 17 def compile(optimizer: nil, loss: nil, metrics: nil) raise "Not implemented" end
evaluate(x, y)
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 25 def evaluate(x, y) raise "Not implemented" end
fit(x, y, epochs: nil)
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 21 def fit(x, y, epochs: nil) raise "Not implemented" end
summary()
click to toggle source
# File lib/tensorflow/keras/models/sequential.rb, line 29 def summary sep = "_________________________________________________________________\n" output_shape = nil @layers.each do |layer| layer.build(output_shape) if layer.respond_to?(:build) output_shape = layer.output_shape end total_params = @layers.map(&:count_params).sum summary = String.new("") summary << "Model: \"sequential\"\n" summary << sep summary << "Layer (type) Output Shape Param # \n" summary << "=================================================================\n" summary << @layers.map { |l| "%-28s %-25s %-10s\n" % [l.class.name.split("::").last, ([nil] + l.output_shape[1..-1]).inspect, l.count_params] }.join(sep) summary << "=================================================================\n" summary << "Total params: #{total_params}\n" summary << "Trainable params: #{total_params}\n" summary << "Non-trainable params: 0\n" summary << sep puts summary end