Cohere Health Interview Question

Implement KNN algorithm. A base class is implemented, you need to complete some of the methods.