Update agent.py

2020-10-22 20:31:50 +08:00
parent 6be9904333
commit 28942780af
1 changed files with 3 additions and 3 deletions
--- a/codes/Q-learning/agent.py
+++ b/codes/Q-learning/agent.py
@@ -84,11 +84,11 @@ class QLearning(object):
    def save(self):
        '''把 Q表格 的数据保存到文件中
        '''
-        npy_file = './Q_table.npy'
+        npy_file = './result/Q_table.npy'
        np.save(npy_file, self.Q_table)
        print(npy_file + ' saved.')
-    def load(self, npy_file='./Q_table.npy'):
+    def load(self, npy_file='./result/Q_table.npy'):
        '''从文件中读取数据到 Q表格
        '''
        self.Q_table = np.load(npy_file)
-        print(npy_file + 'loaded.')
+        print(npy_file + 'loaded.')