Hi @moe18 , Added a new PR for KNN question here. Please review and let me know. Thx! https://github.com/Open-Deep-ML/DML-OpenProblem/pull/514