76 lines
1.3 KiB
Python
76 lines
1.3 KiB
Python
import random
|
|
import gym
|
|
from ai_gym import fh_ac_ai_gym
|
|
from Task.Propositions import *
|
|
from Task.Propositions2 import *
|
|
|
|
#Create Gym environment
|
|
wumpus_env = gym.make('Wumpus-v0', disable_env_checker=True)
|
|
wumpus_env.reset()
|
|
wumpus_env.render()
|
|
|
|
#Define the possible actions
|
|
REVERSE_ACTION_DICT= {0: "WALK", 1:"TURNLEFT", 2:"TURNRIGHT", 3:"GRAB", 4: "SHOOT", 5:"CLIMB"}
|
|
POSSIBLE_ACTIONS = {v: k for k, v in REVERSE_ACTION_DICT.items()}
|
|
|
|
|
|
print("=" * 70)
|
|
|
|
agent = KnowledgebasedAgent("CNF")
|
|
obs = wumpus_env.step(1)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("W21")
|
|
agent.ASK("W12")
|
|
wumpus_env.render()
|
|
|
|
print("=" * 70)
|
|
obs = wumpus_env.step(0)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("W21")
|
|
agent.ASK("W12")
|
|
wumpus_env.render()
|
|
|
|
print("=" * 70)
|
|
obs = wumpus_env.step(0)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("P12")
|
|
agent.ASK("P23")
|
|
agent.ASK("P14")
|
|
wumpus_env.render()
|
|
|
|
print("=" * 70)
|
|
wumpus_env.step(2)
|
|
obs = wumpus_env.step(0)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("P12")
|
|
agent.ASK("P23")
|
|
agent.ASK("P14")
|
|
wumpus_env.render()
|
|
|
|
wumpus_env.reset()
|
|
print("=" * 70)
|
|
print("=" * 70)
|
|
print("=" * 70)
|
|
|
|
|
|
agent = HornKnowledgebasedAgent()
|
|
wumpus_env.step(1)
|
|
obs = wumpus_env.step(2)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("W21")
|
|
agent.ASK("W12")
|
|
wumpus_env.render()
|
|
|
|
print("=" * 70)
|
|
obs = wumpus_env.step(0)
|
|
print(obs)
|
|
agent.TELL(obs)
|
|
agent.ASK("W21")
|
|
agent.ASK("W12")
|
|
wumpus_env.render()
|