Skip to content

Enable use of MLP as summary network #513

@vpratz

Description

@vpratz

The MLP class currently cannot be used as a summary network (AttributeError: 'MLP' object has no attribute 'compute_metrics').

I think it would be good to have MLPs available as general purpose summary networks, when none of the specialized summary networks is suitable for the problem at hand. A light wrapper around MLP would be sufficient.

Open questions:

  1. Do we need a new class? My tendency is yes, because
    a) we might not want to incorporate SummaryNetwork arguments and functions in the general purpose MLP class
    b) the current implementation has an activation after the last layer, which is not desirable for a summary network, so the class should include an additional projector
  2. Naming: How to avoid confusion with the MLP class. The most obvious name would be SummaryMLP.

@stefanradev93 @LarsKue What do you think, is this something we want to include?

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions