25 lines
542 B
Python
25 lines
542 B
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
'''
|
|
@Author: John
|
|
@Email: johnjim0816@gmail.com
|
|
@Date: 2020-06-11 16:30:09
|
|
@LastEditor: John
|
|
@LastEditTime: 2020-06-11 22:27:24
|
|
@Discription:
|
|
@Environment: python 3.7.7
|
|
'''
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import os
|
|
|
|
def plot(item,ylabel='rewards'):
|
|
plt.figure()
|
|
plt.plot(np.arange(len(item)), item)
|
|
plt.title(ylabel+' of CnnDQN')
|
|
plt.ylabel('rewards')
|
|
plt.xlabel('episodes')
|
|
|
|
plt.savefig(os.path.dirname(__file__)+"/result/"+ylabel+".png")
|
|
plt.show()
|