add dqn

codes/dqn/plot.py

@@ -5,19 +5,22 @@
 @Email: johnjim0816@gmail.com
 @Date: 2020-06-11 16:30:09
 @LastEditor: John
-@LastEditTime: 2020-06-14 11:38:42
+LastEditTime: 2020-08-20 16:34:34
 @Discription: 
 @Environment: python 3.7.7
 '''
 import matplotlib.pyplot as plt
+import pandas as pd
+import seaborn as sns
 import numpy as np
 import os 
 
 def plot(item,ylabel='rewards'):
+    sns.set()
     plt.figure()
     plt.plot(np.arange(len(item)), item)
     plt.title(ylabel+' of DQN') 
-    plt.ylabel('rewards')
+    plt.ylabel(ylabel)
     plt.xlabel('episodes')
     plt.savefig(os.path.dirname(__file__)+"/result/"+ylabel+".png")
     plt.show()