Skip to content

Implement Sparsemax Activation and Loss #21

@seanpmorgan

Description

@seanpmorgan

Per the RFC, we need to move sparsemax from contrib to addons:

  • Moving sparsemax to addons/activations and modifying test cases to run in TF2.
  • Restructuring sparsemax_loss in an OO format by inheriting from base Keras Loss, modifying the code to match those APIs, and modifying test cases to run in TF2.

https://arxiv.org/abs/1602.02068

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions