Method save

modelzoo/features/embedding_variable/dien/script/model.py:331–333

(self, sess, path)

            return probs, loss, accuracy, 0

    def save(self, sess, path):
        saver = tf.train.Saver()
        saver.save(sess, save_path=path)

    def summary(self, sess, path, iter, hist):
        writer = tf.summary.FileWriter(path, sess.graph)