faris-k / fastsiam-wafers

Self-Supervised Representation Learning of Wafer Maps with FastSiam
MIT License
6 stars 0 forks source link

knn_k isn't being used anywhere 😢😿 #5

Closed faris-k closed 1 year ago

faris-k commented 1 year ago

Benchmarks were all run using knn_k=200, the default value in KNNBenchmarkModule. No wonder accuracy/F1 scores never went higher than 0.5-0.6 ☹️. Note that for kNN classifiers, choosing a values of $k$ this high could make it impossible to predict minority classes, which I highly suspect is what's going on here.

Tasks: