149 lines
6.0 KiB
Python
149 lines
6.0 KiB
Python
# valueIterationAgents.py
|
|
# -----------------------
|
|
# Licensing Information: You are free to use or extend these projects for
|
|
# educational purposes provided that (1) you do not distribute or publish
|
|
# solutions, (2) you retain this notice, and (3) you provide clear
|
|
# attribution to UC Berkeley, including a link to http://ai.berkeley.edu.
|
|
#
|
|
# Attribution Information: The Pacman AI projects were developed at UC Berkeley.
|
|
# The core projects and autograders were primarily created by John DeNero
|
|
# (denero@cs.berkeley.edu) and Dan Klein (klein@cs.berkeley.edu).
|
|
# Student side autograding was added by Brad Miller, Nick Hay, and
|
|
# Pieter Abbeel (pabbeel@cs.berkeley.edu).
|
|
|
|
|
|
# valueIterationAgents.py
|
|
# -----------------------
|
|
# Licensing Information: You are free to use or extend these projects for
|
|
# educational purposes provided that (1) you do not distribute or publish
|
|
# solutions, (2) you retain this notice, and (3) you provide clear
|
|
# attribution to UC Berkeley, including a link to http://ai.berkeley.edu.
|
|
#
|
|
# Attribution Information: The Pacman AI projects were developed at UC Berkeley.
|
|
# The core projects and autograders were primarily created by John DeNero
|
|
# (denero@cs.berkeley.edu) and Dan Klein (klein@cs.berkeley.edu).
|
|
# Student side autograding was added by Brad Miller, Nick Hay, and
|
|
# Pieter Abbeel (pabbeel@cs.berkeley.edu).
|
|
|
|
|
|
import mdp, util
|
|
|
|
from learningAgents import ValueEstimationAgent
|
|
import collections
|
|
|
|
class ValueIterationAgent(ValueEstimationAgent):
|
|
"""
|
|
* Please read learningAgents.py before reading this.*
|
|
|
|
A ValueIterationAgent takes a Markov decision process
|
|
(see mdp.py) on initialization and runs value iteration
|
|
for a given number of iterations using the supplied
|
|
discount factor.
|
|
"""
|
|
def __init__(self, mdp: mdp.MarkovDecisionProcess, discount = 0.9, iterations = 100):
|
|
"""
|
|
Your value iteration agent should take an mdp on
|
|
construction, run the indicated number of iterations
|
|
and then act according to the resulting policy.
|
|
|
|
Some useful mdp methods you will use:
|
|
mdp.getStates()
|
|
mdp.getPossibleActions(state)
|
|
mdp.getTransitionStatesAndProbs(state, action)
|
|
mdp.getReward(state, action, nextState)
|
|
mdp.isTerminal(state)
|
|
"""
|
|
self.mdp = mdp
|
|
self.discount = discount
|
|
self.iterations = iterations
|
|
self.values = util.Counter() # A Counter is a dict with default 0
|
|
self.runValueIteration()
|
|
|
|
def runValueIteration(self):
|
|
"""
|
|
Run the value iteration algorithm. Note that in standard
|
|
value iteration, V_k+1(...) depends on V_k(...)'s.
|
|
"""
|
|
"*** YOUR CODE HERE ***"
|
|
# Write value iteration code here
|
|
# Hints:
|
|
# - After each iteration, store the new values in self.values
|
|
# - When updating a value, use self.values[state] = <new value>
|
|
# - You will need to copy the state values into a separate dictionary
|
|
# to avoid changing values before computing the update.
|
|
# - The difference between the new value and the old value (|V_k+1(s) - V_k(s)|)
|
|
# should be less than self.epsilon for all states s
|
|
# - Make sure to use the discount factor self.discount
|
|
# - Make sure to use the bellman equations to update the state values
|
|
# - The number of iterations is given by self.iterations
|
|
# - You may use the util.Counter() class
|
|
# - You may also use the self.mdp.getTransitionStatesAndProbs(state, action) method
|
|
# - You may also use the self.mdp.getReward(state, action, nextState) method
|
|
# - You may also use the self.mdp.getPossibleActions(state) method
|
|
# - You may also use the self.mdp.isTerminal(state) method
|
|
|
|
for i in range(self.iterations):
|
|
newValues = util.Counter()
|
|
for state in self.mdp.getStates():
|
|
if self.mdp.isTerminal(state):
|
|
newValues[state] = 0
|
|
else:
|
|
maxQValue = float("-inf")
|
|
for action in self.mdp.getPossibleActions(state):
|
|
qValue = self.computeQValueFromValues(state, action)
|
|
maxQValue = max(maxQValue, qValue)
|
|
newValues[state] = maxQValue
|
|
self.values = newValues
|
|
|
|
|
|
def getValue(self, state):
|
|
"""
|
|
Return the value of the state (computed in __init__).
|
|
"""
|
|
return self.values[state]
|
|
|
|
def computeQValueFromValues(self, state, action):
|
|
"""
|
|
Compute the Q-value of action in state from the
|
|
value function stored in self.values.
|
|
"""
|
|
"*** YOUR CODE HERE ***"
|
|
qValue = 0
|
|
for nextState, prob in self.mdp.getTransitionStatesAndProbs(state, action):
|
|
reward = self.mdp.getReward(state, action, nextState)
|
|
qValue += prob * (reward + self.discount * self.values[nextState])
|
|
return qValue
|
|
|
|
|
|
def computeActionFromValues(self, state):
|
|
"""
|
|
The policy is the best action in the given state
|
|
according to the values currently stored in self.values.
|
|
|
|
You may break ties any way you see fit. Note that if
|
|
there are no legal actions, which is the case at the
|
|
terminal state, you should return None.
|
|
"""
|
|
"*** YOUR CODE HERE ***"
|
|
bestAction = None
|
|
bestQValue = float("-inf")
|
|
for action in self.mdp.getPossibleActions(state):
|
|
qValue = self.computeQValueFromValues(state, action)
|
|
if qValue > bestQValue:
|
|
bestQValue = qValue
|
|
bestAction = action
|
|
return bestAction
|
|
|
|
|
|
def getPolicy(self, state):
|
|
return self.computeActionFromValues(state)
|
|
|
|
def getAction(self, state):
|
|
"Returns the policy at the state (no exploration)."
|
|
return self.computeActionFromValues(state)
|
|
|
|
def getQValue(self, state, action):
|
|
return self.computeQValueFromValues(state, action)
|
|
|
|
|