add some codes

codes/dqn_cnn/plot.py

@@ -0,0 +1,24 @@
+#!/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()