All Projects → overshiki → Kmeans_pytorch

overshiki / Kmeans_pytorch

pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Kmeans pytorch

Aiopen
AIOpen是一个按人工智能三要素(数据、算法、算力)进行AI开源项目分类的汇集项目,项目致力于跟踪目前人工智能(AI)的深度学习(DL)开源项目,并尽可能地罗列目前的开源项目,同时加入了一些曾经研究过的代码。通过这些开源项目,使初次接触AI的人们对人工智能(深度学习)有更清晰和更全面的了解。
Stars: ✭ 62 (-23.46%)
Mutual labels:  gpu
Vectorsinsearch
Dice.com repo to accompany the dice.com 'Vectors in Search' talk by Simon Hughes, from the Activate 2018 search conference, and the 'Searching with Vectors' talk from Haystack 2019 (US). Builds upon my conceptual search and semantic search work from 2015
Stars: ✭ 71 (-12.35%)
Mutual labels:  kmeans
Phenomenon Px
⚡️ The fastest way to create pixel shaders.
Stars: ✭ 77 (-4.94%)
Mutual labels:  gpu
Ggnn
GGNN: State of the Art Graph-based GPU Nearest Neighbor Search
Stars: ✭ 63 (-22.22%)
Mutual labels:  gpu
Cloudml
R interface to Google Cloud Machine Learning Engine
Stars: ✭ 66 (-18.52%)
Mutual labels:  gpu
Vfio
A script I've made to make passing PCI devices to a VM less of a chore. I use it for single-gpu VM gaming and other PCI/LiveCD/PXE/VM/RawImage testing given the script's accessibility.
Stars: ✭ 71 (-12.35%)
Mutual labels:  gpu
How To Mine Newsfeed Data And Extract Interactive Insights In Python
A practical guide to topic mining and interactive visualizations
Stars: ✭ 61 (-24.69%)
Mutual labels:  kmeans
Cuda Design Patterns
Some CUDA design patterns and a bit of template magic for CUDA
Stars: ✭ 78 (-3.7%)
Mutual labels:  gpu
Rpi Vk Driver
VK driver for the Raspberry Pi (Broadcom Videocore IV)
Stars: ✭ 1,160 (+1332.1%)
Mutual labels:  gpu
Cekirdekler
Multi-device OpenCL kernel load balancer and pipeliner API for C#. Uses shared-distributed memory model to keep GPUs updated fast while using same kernel on all devices(for simplicity).
Stars: ✭ 76 (-6.17%)
Mutual labels:  gpu
Vbiosfinder
Extract embedded VBIOS from (almost) any BIOS Update
Stars: ✭ 64 (-20.99%)
Mutual labels:  gpu
Nx
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
Stars: ✭ 1,133 (+1298.77%)
Mutual labels:  gpu
Compute
A C++ GPU Computing Library for OpenCL
Stars: ✭ 1,192 (+1371.6%)
Mutual labels:  gpu
Tsne Cuda
GPU Accelerated t-SNE for CUDA with Python bindings
Stars: ✭ 1,120 (+1282.72%)
Mutual labels:  gpu
Nplusminer
NPlusMiner + GUI | NVIDIA/AMD/CPU miner | AI | Autoupdate | MultiRig remote management
Stars: ✭ 75 (-7.41%)
Mutual labels:  gpu
Pycuda
CUDA integration for Python, plus shiny features
Stars: ✭ 1,112 (+1272.84%)
Mutual labels:  gpu
Parenchyma
An extensible HPC framework for CUDA, OpenCL and native CPU.
Stars: ✭ 71 (-12.35%)
Mutual labels:  gpu
Aardvark.rendering
The dependency-aware, high-performance aardvark rendering engine. This repo is part of aardvark - an open-source platform for visual computing, real-time graphics and visualization.
Stars: ✭ 79 (-2.47%)
Mutual labels:  gpu
Hyperlearn
50% faster, 50% less RAM Machine Learning. Numba rewritten Sklearn. SVD, NNMF, PCA, LinearReg, RidgeReg, Randomized, Truncated SVD/PCA, CSR Matrices all 50+% faster
Stars: ✭ 1,204 (+1386.42%)
Mutual labels:  gpu
Cudart.jl
Julia wrapper for CUDA runtime API
Stars: ✭ 75 (-7.41%)
Mutual labels:  gpu

pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support

usage:

from kmeans_pytorch.kmeans import lloyd
import numpy as np 

A = np.concatenate([np.random.randn(1000, 2), p.random.randn(1000, 2)+3, p.random.randn(1000, 2)+6], axis=0)

#lloyd(X, n_clusters, device=0, tol=1e-4)
clusters_index, centers = lloyd(A, 2, device=0, tol=1e-4)

See kmeans_test.ipynb for some test examples

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].