dqn利用gym的例子

最新推荐文章于 2024-11-13 17:24:18 发布

andyzou221312

最新推荐文章于 2024-11-13 17:24:18 发布

阅读量365

点赞数

分类专栏： torch 文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/andyzou221312/article/details/126591358

版权

本文详细介绍了如何使用Python和深度学习库，通过Deep Q-Network（DQN）算法在OpenAI Gym环境中进行强化学习的实践案例。通过对环境的交互和策略迭代，DQN成功地学会了游戏策略。

摘要由CSDN通过智能技术生成


"""
View more, visit my tutorial page: https://mofanpy.com/tutorials/
My Youtube Channel: https://www.youtube.com/user/MorvanZhou
More about Reinforcement learning: https://mofanpy.com/tutorials/machine-learning/reinforcement-learning/

Dependencies:
torch: 0.4
gym: 0.8.1
numpy
"""
import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
import gym

# Hyper Parameters
BATCH_SIZE = 32
LR = 0.01                   # learning rate
贪婪阈值 = 0.9               # greedy policy
GAMMA = 0.9                 # reward discount
TARGET_REPLACE_ITER = 100   # target update frequency
MEMORY_CAPACITY = 2000
env = gym.make('CartPole-v0')
env = env.unwrapped
N_ACTIONS = env.action_space.n
N_STATES = env.observation_space.shape[0]
ENV_A_SHAPE = 0 if isinstance(env.action_space.sample(), int) else env.action_space.sample().shape     # to confirm the shape


class Net(nn.Module):
    def __init_