LogStatsWriterTensorboard

class maze.core.log_stats.log_stats_writer_tensorboard.LogStatsWriterTensorboard(log_dir: str, tensorboard_render_figure: bool)

Log statistics writer implementation for Tensorboard. :param log_dir: log_dir for TensorBoard :param tensorboard_render_figure: Indicates whether to visualize the actions taken in TensorBoard.

close() → None

(overrides LogStatsWriter)

LogStatsWriter.write implementation

on_log_step_increment()

Hooked into increment_log_step, called by the logging system immediately before the increment.

write(path: str, step: int, stats: Dict[maze.core.log_stats.log_stats.LogStatsKey, Union[int, float, numpy.ndarray, dict]]) → None

(overrides LogStatsWriter)

LogStatsWriter.write implementation