EdgeML
EdgeML copied to clipboard
K-Median/Mean heuristic for ProtoNN missing in Pytorch
- 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 :/ .