class NeuralNetCritic:
def __init__(self):
self.net = NeuralNet(nodes=config["nodes"])
def getTDError(self, state, newState, reinforcement):
return self.net.criterion(state, newState, reinforcement)
def updateEligibility(self, state, isCurrentState=False):
return True
def updateValueFunction(self, state, TDError):
self.net.train(state, TDError)
