Skip to content

TypeError: expand_dims() got an unexpected keyword argument 'dim' #2

Description

@forhonourlx

def dueling_dqn(input_shape, action_size, learning_rate):
...
state_value = Lambda(lambda s: K.expand_dims(s[:, 0], dim=-1), output_shape=(action_size,))(state_value)
...

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions