Source code for draugr.torch_utilities.writers.visdom.plotting

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = "Christian Heider Nielsen"

import numpy
import visdom

vis = visdom.Visdom()

__all__ = ["plot_episode_stats"]


[docs]def plot_episode_stats(stats): """ :param stats: :type stats: :return: :rtype:""" vis.line( X=numpy.arange(len(stats.signal_mas)), Y=numpy.array(stats.signal_mas), win="DDPG MEAN REWARD (100 episodes)", opts=dict( title=("DDPG MEAN REWARD (100 episodes)"), ylabel="MEAN REWARD (100 episodes)", xlabel="Episode", ), ) # Plot the mean of last 100 episode rewards over time. vis.line( X=numpy.cumsum(stats.episode_lengths), Y=numpy.arange(len(stats.episode_lengths)), win="DDPG Episode per time step", opts=dict( title=("DDPG Episode per time step"), ylabel="Episode", xlabel="Time Steps" ), ) # Plot time steps and episode number.