bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:datawhalechina/leedeeprl-notes

Browse Source
This commit is contained in:
qiwang067
2020-10-22 21:40:40 +08:00
parent 092aed74ac 28942780af
commit ef9c34a978
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
codes/Q-learning/agent.py
View File

@@ -84,10 +84,10 @@ class QLearning(object):
def save(self): def save(self):
'''把 Q表格 的数据保存到文件中 '''把 Q表格 的数据保存到文件中
''' '''
npy_file = './Q_table.npy' npy_file = './result/Q_table.npy'
np.save(npy_file, self.Q_table) np.save(npy_file, self.Q_table)
print(npy_file + ' saved.') print(npy_file + ' saved.')
def load(self, npy_file='./Q_table.npy'): def load(self, npy_file='./result/Q_table.npy'):
'''从文件中读取数据到 Q表格 '''从文件中读取数据到 Q表格
''' '''
self.Q_table = np.load(npy_file) self.Q_table = np.load(npy_file)