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