An open API service indexing awesome lists of open source software.

https://github.com/mujadded/knn_with_pytorch


https://github.com/mujadded/knn_with_pytorch

Last synced: 25 days ago
JSON representation

Awesome Lists containing this project

README

          

# knn_with_pytorch

A Simple implementation of KNN using pyTorch.

## Files

`knn.py` has the main code for K-Nearest Neighbor

`knn_test.ipynb` has the playgroud where this is tested.

Example of KNN:

![KNN plot](https://github.com/Mujadded/knn_with_pytorch/blob/main/output.png?raw=true)