• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

readchar api 이용한 방법입니다.

18.07.21 18:48 작성 조회수 276

0

아래 설치

pip install readchar

아래 코드 실행

import gym

from gym.envs.registration import register

import readchar

LEFT = 0

DOWN = 1

RIGHT = 2

UP = 3

arrow_keys = {

'x1b[A' : UP,

'x1b[B' : DOWN,

'x1b[C' : RIGHT,

'x1b[D' : LEFT

}

register(

id='FrozenLake-v3',

entry_point="gym.envs.toy_text:FrozenLakeEnv",

kwargs={'map_name':'4x4','is_slippery':False})

env = gym.make("FrozenLake-v3")

env.render() #환경을 화면으로 출력

while True:

key = readchar.readkey() #키보드 입력을 받는다

if key not in arrow_keys.keys():

print("Game aborted!")

break

action = arrow_keys[key] #에이젼트의 움직임

state, reward, done, info = env.step(action) #움직임에 따른 결과값들

env.render() #화면을 다시 출력

print("State:", state, "Action", action, "Reward:", reward, "Info:", info)

if done: #도착하면 게임을 끝낸다.

print("Finished with reward", reward)

break

답변 0

답변을 작성해보세요.

답변을 기다리고 있는 질문이에요.
첫번째 답변을 남겨보세요!