keras 如何将两个数组作为Gym.Env的输出以适合DQN NN

k5ifujac  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(148)

不知道如何制作健身房。Env提供了两个独立的数组。它只是似乎将它们组合成一个包含两个数组的数组。但符合DQN NN要求两个数组。我希望将两个数组分别放入NN。
我试着展示尽可能多的代码,但是代码太多了。
我试着在观察空间里玩了一会儿,试了盒子和元组,似乎找不到我哪里出错了。
第一个

z9gpfhce

z9gpfhce1#

好的,我找到了一个解决办法。我希望这对任何有类似问题的人都有帮助,因为关于最初的问题似乎没有太多的信息。
我将数据作为单个数组输入。

  1. data_input = Input(shape=(500, 3, 11))

然后我就可以使用tf.splt()在模型本身中拆分输入

  1. min1, min15, acc = tf.split(data_input, num_or_size_splits=3, axis=2)

然后通过不同的网络运行它们,然后用concatenate将它们组合在一起

  1. z = concatenate([acc, combined])
  2. z = Flatten()(z)
  3. z = Dense(512, activation='swish'(z)
  4. z = Dense(512, activation='swish'(z)
  5. z = Dense(3, activation='softmax')(z)
  6. model = keras.Model(inputs=data_input, outputs=z)
  7. return model
展开查看全部

相关问题