Skip to content

Table based agent #1348

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Vance-John wants to merge 5 commits into from
Closed

Table based agent #1348

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
883f0fe
Adding work we did in class yesterday
Sep 2, 2026
3a3a955
Expand trivial vacuum world from 1x2 to 2x2.
Sep 4, 2026
628828d
Merge pull request #1 from Vance-John/2x2-trivial-vacuum
Ldbonequi Sep 4, 2026
c0e26a0
change model based agent to work in 2 dimensions
Ldbonequi Sep 4, 2026
7275d93
Adding table based agent
Sep 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
138 changes: 120 additions & 18 deletions aima/agents.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def rule_match(state, rules):
# ______________________________________________________________________________


loc_A, loc_B = (0, 0), (1, 0) # The two locations for the Vacuum world
loc_A, loc_B, loc_C, loc_D = (0, 0), (1, 0), (0, 1), (1, 1) # The two locations for the Vacuum world


def RandomVacuumAgent():
Expand All @@ -203,7 +203,7 @@ def RandomVacuumAgent():
>>> environment.status == {(1,0):'Clean' , (0,0) : 'Clean'}
True
"""
return Agent(RandomAgentProgram(['Right', 'Left', 'Suck', 'NoOp']))
return Agent(RandomAgentProgram(['Right', 'Left', 'Up', 'Down', 'Suck', 'NoOp']))


def TableDrivenVacuumAgent():
Expand All @@ -215,16 +215,84 @@ def TableDrivenVacuumAgent():
>>> environment.status == {(1,0):'Clean' , (0,0) : 'Clean'}
True
"""
table = {((loc_A, 'Clean'),): 'Right',
((loc_A, 'Dirty'),): 'Suck',
((loc_B, 'Clean'),): 'Left',
((loc_B, 'Dirty'),): 'Suck',
((loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
((loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_B, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_B, 'Dirty'), (loc_B, 'Clean')): 'Left',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_A, 'Dirty')): 'Suck'}
table = {
((loc_A, 'Dirty'),): 'Suck',
((loc_A, 'Clean'),): 'Right',
((loc_B, 'Dirty'),): 'Suck',
((loc_B, 'Clean'),): 'Down',
((loc_D, 'Dirty'),): 'Suck',
((loc_D, 'Clean'),): 'Left',
((loc_C, 'Dirty'),): 'Suck',
((loc_C, 'Clean'),): 'Up',

((loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
((loc_B, 'Dirty'), (loc_B, 'Clean')): 'Down',
((loc_D, 'Dirty'), (loc_D, 'Clean')): 'Left',
((loc_C, 'Dirty'), (loc_C, 'Clean')): 'Up',
((loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',
((loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',

((loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_A, 'Clean'), (loc_B, 'Dirty'), (loc_B, 'Clean')): 'Down',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',
((loc_B, 'Clean'), (loc_D, 'Dirty'), (loc_D, 'Clean')): 'Left',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_C, 'Clean'), (loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
((loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',
((loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',
((loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_D, 'Clean'), (loc_C, 'Dirty'), (loc_C, 'Clean')): 'Up',
((loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',

((loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',
((loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Dirty'), (loc_D, 'Clean')): 'Left',
((loc_A, 'Clean'), (loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_A, 'Clean'), (loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Dirty'), (loc_B, 'Clean')): 'Down',
((loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',
((loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Dirty'), (loc_C, 'Clean')): 'Up',
((loc_B, 'Clean'), (loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_B, 'Clean'), (loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Clean')): 'Up',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Clean'), (loc_C, 'Dirty')): 'Suck',
((loc_B, 'Dirty'), (loc_B, 'Clean'), (loc_D, 'Dirty'), (loc_D, 'Clean')): 'Left',
((loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Clean')): 'Left',
((loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Clean'), (loc_D, 'Dirty')): 'Suck',
((loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Dirty'), (loc_B, 'Clean')): 'Down',
((loc_C, 'Clean'), (loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_C, 'Clean'), (loc_A, 'Dirty'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
((loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Clean')): 'Down',
((loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Clean'), (loc_B, 'Dirty')): 'Suck',
((loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Dirty'), (loc_A, 'Clean')): 'Right',
((loc_D, 'Clean'), (loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',
((loc_D, 'Clean'), (loc_C, 'Dirty'), (loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Clean')): 'Right',
((loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Clean'), (loc_A, 'Dirty')): 'Suck',
((loc_D, 'Dirty'), (loc_D, 'Clean'), (loc_C, 'Dirty'), (loc_C, 'Clean')): 'Up',

}
return Agent(TableDrivenAgentProgram(table))


Expand All @@ -247,7 +315,11 @@ def program(percept):
elif location == loc_A:
return 'Right'
elif location == loc_B:
return 'Down'
elif location == loc_D:
return 'Left'
elif location == loc_C:
return 'Up'

return Agent(program)

Expand All @@ -261,20 +333,24 @@ def ModelBasedVacuumAgent():
>>> environment.status == {(1,0):'Clean' , (0,0) : 'Clean'}
True
"""
model = {loc_A: None, loc_B: None}
model = {loc_A: None, loc_B: None, loc_C: None, loc_D: None}

def program(percept):
"""Same as ReflexVacuumAgent, except if everything is clean, do NoOp."""
location, status = percept
model[location] = status # Update the model here
if model[loc_A] == model[loc_B] == 'Clean':
if model[loc_A] == model[loc_B] == model[loc_C] == model[loc_D] == 'Clean':
return 'NoOp'
elif status == 'Dirty':
return 'Suck'
elif location == loc_A:
return 'Right'
elif location == loc_B:
return 'Down'
elif location == loc_D:
return 'Left'
elif location == loc_C:
return 'Up'

return Agent(program)

Expand Down Expand Up @@ -807,7 +883,9 @@ class TrivialVacuumEnvironment(Environment):
def __init__(self):
super().__init__()
self.status = {loc_A: random.choice(['Clean', 'Dirty']),
loc_B: random.choice(['Clean', 'Dirty'])}
loc_B: random.choice(['Clean', 'Dirty']),
loc_C: random.choice(['Clean', 'Dirty']),
loc_D: random.choice(['Clean', 'Dirty'])}

def thing_classes(self):
"""Return the Thing/Agent classes that may populate this vacuum world."""
Expand All @@ -820,20 +898,44 @@ def percept(self, agent):
def execute_action(self, agent, action):
"""Change agent's location and/or location's status; track performance.
Score 10 for each dirt cleaned; -1 for each move."""
if action == 'Right':

if action == 'Right' and agent.location == (0,0):
agent.location = loc_B
agent.performance -= 1
elif action == 'Left':
elif action == 'Right' and agent.location == (0,1):
agent.location = loc_D
agent.performance -= 1
elif action == 'Left' and agent.location == (1,0):
agent.location = loc_A
agent.performance -= 1
elif action == 'Left' and agent.location == (1,1):
agent.location = loc_C
agent.performance -= 1
elif action == 'Down' and agent.location == (0,0):
agent.location = loc_C
agent.performance -= 1
elif action == 'Down' and agent.location == (1,0):
agent.location = loc_D
agent.performance -= 1
elif action == 'Up' and agent.location == (0,1):
agent.location = loc_A
agent.performance -= 1
elif action == 'Up' and agent.location == (1,1):
agent.location = loc_B
agent.performance -= 1

elif action == 'Suck':
if self.status[agent.location] == 'Dirty':
agent.performance += 10
self.status[agent.location] = 'Clean'

## Should the agent be penalized for doing an invalid action?
## For example if it chooses Left while its already at loc_A, should it still get agent.performance -= 1
## - Carter

def default_location(self, thing):
"""Agents start in either location at random."""
return random.choice([loc_A, loc_B])
return random.choice([loc_A, loc_B, loc_C, loc_D])


# ______________________________________________________________________________
Expand Down
Loading