Skip to content

K-Median/Mean heuristic for ProtoNN missing in Pytorch #138

Description

  • The median heuristic function utils.medianHeuristic is missing in the pytorch branch. Please copy it to edgeml_pytorch.utils.
  • It is being called from here when you run the example. That is, if gamma is not provided, we will estimate it.
  • As can be seen from the Notes here and the code here, the implementation is unaware of class labels and hence per-class k-means is not there. This requires effort :/ .

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions