module TensorStream::Trainer

Public Class Methods

write_graph(graph, path, filename, as_text: true, serializer: :yaml) click to toggle source
# File lib/tensor_stream/trainer.rb, line 18
def self.write_graph(graph, path, filename, as_text: true, serializer: :yaml)
  raise "only supports as_text=true for now" unless as_text

  serializer = constantize("TensorStream::#{camelize(serializer.to_s)}") if serializer.is_a?(Symbol)

  new_filename = File.join(path, filename)
  serializer.new.get_string(graph).tap do |str|
    File.write(new_filename, str)
  end
end