VinF / deer

DEEp Reinforcement learning framework
Other
485 stars 126 forks source link

Update NN_keras.py #81

Closed aaspeel closed 3 years ago

aaspeel commented 3 years ago

In the function inputDimensions(self) in the Environment class, we fix a bug when the shape of a scalar is explicitly given. For example, returning [(5,1)] was not allowed because [(5,)] is expected. This is fixed, which allows to return [(5,1)] or [(5,)]. Therefore, returning [(5, n)] will work for n == 1 and for n> 1.