lib/tensorflow/keras/models/sequential.rb in tensorflow-0.1.2 vs lib/tensorflow/keras/models/sequential.rb in tensorflow-0.2.0

@@ -23,9 +23,34 @@
         end
 
         def evaluate(x, y)
           raise "Not implemented"
         end
+
+        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
       end
     end
   end
 end