Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/rmiguelkelly/quickcluster
A KMeans implemented in C++ with Python bindings and GPU acceleration
https://github.com/rmiguelkelly/quickcluster
clustering clustering-algorithm cpp cuda gpu kmeans kmeans-clustering metal objective-c python python3 unsupervised-learning
Last synced: 4 months ago
JSON representation
A KMeans implemented in C++ with Python bindings and GPU acceleration
- Host: GitHub
- URL: https://github.com/rmiguelkelly/quickcluster
- Owner: rmiguelkelly
- License: mit
- Created: 2024-04-11T20:13:56.000Z (10 months ago)
- Default Branch: main
- Last Pushed: 2024-07-20T19:52:48.000Z (6 months ago)
- Last Synced: 2024-09-30T02:21:16.637Z (4 months ago)
- Topics: clustering, clustering-algorithm, cpp, cuda, gpu, kmeans, kmeans-clustering, metal, objective-c, python, python3, unsupervised-learning
- Language: C++
- Homepage:
- Size: 65.4 KB
- Stars: 6
- Watchers: 1
- Forks: 0
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# QuickCluster 🔥
A KMeans algorithm implemented in C++ along with Pythoin bindings that runs on the CPU or the GPU.
I wrote this project more as a proof of concept for some key learning points I wanted to understand.
This project is definitely still in beta so there may be some bugs and upgrades to add as time goes on.
## Features
- Simple Python interface similar to SKLearn's KMeans
- Dynamic libraries for MacOS and Linux (Windows coming soon) for C++ integration
- Support for Apple's GPU for hardware acceleration (CUDA coming soon too)
## Building
Build the dynamic library from source
```bash
git clone https://github.com/rmiguelkelly/QuickCluster
cd QuickCluster/quickcluster
```
MacOS
```bash
make darwin
```
Linux
```bash
make linux
```
Windows and CUDA implementations
```bash
Coming soon!
```
## Examples
Some example code for running the program
Python (CPU) [Example](examples/kmeans-example-cpu.ipynb)
Python (GPU) [Example](examples/kmeans-example-gpu.ipynb)
C++ (CPU) [Example](examples/kmeans-example-cpu.cc)
C++ (GPU) [Example](examples/kmeans-example-gpu.cc)
## Python Installation
You must build the project first with the above steps
```bash
cd ..
pip install .
```
Note: you may have to edit the runtime paths using a tool like `install_name_tool` on MacOS for the dynamic library so that the Python module can link it properly
## Authors
- [@rmiguelkelly](https://www.github.com/rmiguelkelly)