def init_evaluator(self, load_ckpt_path=""):
if load_ckpt_path == "":
load_ckpt_path = self.checkpoint_dir
checkpoint = tf.train.latest_checkpoint(load_ckpt_path)
q_values = self.graph_ops["q_values"]
self.evaluator = tflearn.Evaluator(q_values, checkpoint)
def build_evaluator(trainer, target):
evaluator = tflearn.Evaluator(target, session=trainer.session)
return evaluator
def return_evaluator(self, trainer):
evaluator = tflearn.Evaluator([self.mu, self.logvar],
session=trainer.session)
return evaluator
def define_evaluator(trainer, mu, logvar):
evaluator = tflearn.Evaluator([mu, logvar], session=trainer.session)
return evaluator