29 lines
877 B
Python
29 lines
877 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
class buffer(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.states = []
|
||
|
|
self.actions = []
|
||
|
|
self.rewards = []
|
||
|
|
print("√√√√√Buffer Initialized Success√√√√√")
|
||
|
|
def clearBuffer(self):
|
||
|
|
self.states = []
|
||
|
|
self.actions = []
|
||
|
|
self.rewards = []
|
||
|
|
def getStates(self):
|
||
|
|
return np.asarray(self.states)
|
||
|
|
def getActions(self):
|
||
|
|
return np.asarray(self.actions)
|
||
|
|
def getRewards(self):
|
||
|
|
return np.asarray(self.rewards)
|
||
|
|
|
||
|
|
def saveState(self,state):
|
||
|
|
self.states.append(state)
|
||
|
|
def saveAction(self,action):
|
||
|
|
self.actions.append(action)
|
||
|
|
def saveReward(self,reward):
|
||
|
|
self.rewards.append(reward)
|
||
|
|
def saveBuffers(self,state,action,reward):
|
||
|
|
self.states.append(state)
|
||
|
|
self.actions.append(action)
|
||
|
|
self.rewards.append(reward)
|