Aimbot-PPO/Aimbot-PPO-Python/buffer.py

29 lines
877 B
Python
Raw Normal View History

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)