plot, = plt.plot(ucs_levels, linewidth=2, color=color)
total_plots.append(plot)
counter += 1
plt.legend(total_plots, agent_labels, loc=2)
plt.xlabel('Games')
plt.ylabel('Number of Iterations')
plt.title('foobar')
plt.show()
if __name__ == '__main__':
parser = LogParser()
results = parser.parse_logs()
plotter = Plotter()
for game_set, games in results.iteritems():
plotter.plot_quality_lines(games)
plotter.plot_balance_lines(games)
plotter.bar_plot_days_stayed(games)
plotter.bar_plot_days_stayed_vs_total(games)
plotter.bar_plot_ucs_levels(games)
plotter.bar_plot_qualities(games)
# if game_set == 3:
# game = games[1]
# title = "Quality Rating of Agents - Game 10"
# plotter.plot_quality_lines([game], title)
# title = "Bank Balance of Agents - Game 10"
# plotter.plot_balance_lines([game], title)
