利用Keras-rl配合tensorflow训练深度强化模型遇到问题
报错。。。
如何插入一段漂亮的代码片
去博客设置页面,选择一款你喜欢的代码片高亮样式,下面展示同样高亮的 代码片
.
import numpy as np
import gym
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Activation, Flatten
from tensorflow.keras.optimizers import Adam
from rl.memory import EpisodeParameterMemory
from rl.agents.cem import CEMAgent
ENV_NAME = 'CartPole-v0'
# Get the environment and extract the number of actions.
env = gym.make