19 lines
468 B
Python
19 lines
468 B
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
'''
|
|
Author: John
|
|
Email: johnjim0816@gmail.com
|
|
Date: 2020-11-22 23:23:10
|
|
LastEditor: John
|
|
LastEditTime: 2020-11-23 11:55:24
|
|
Discription:
|
|
Environment:
|
|
'''
|
|
import gym
|
|
|
|
def env_init():
|
|
env = gym.make('CartPole-v0') # 可google为什么unwrapped gym，此处一般不需要
|
|
env.seed(1) # 设置env随机种子
|
|
state_dim = env.observation_space.shape[0]
|
|
n_actions = env.action_space.n
|
|
return env,state_dim,n_actions |