EdgeML icon indicating copy to clipboard operation
EdgeML copied to clipboard

K-Median/Mean heuristic for ProtoNN missing in Pytorch

Open metastableB opened this issue 6 years ago • 0 comments

  • 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 :/ .

metastableB avatar Oct 13 '19 19:10 metastableB