#!/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()