【文件属性】:
文件名称:kmeans_pytorch:使用PyTorch的kmeans
文件大小:961KB
文件格式:ZIP
更新时间:2021-02-04 03:35:30
jekyll github-pages docs gpu pytorch
使用PyTorch的K均值
kmeans的PyTorch实现以利用GPU
入门
import torch
import numpy as np
from kmeans_pytorch import kmeans
# data
data_size, dims, num_clusters = 1000, 2, 3
x = np.random.randn(data_size, dims) / 6
x = torch.from_numpy(x)
# kmeans
cluster_ids_x, cluster_centers = kmeans(
X=x, num_clusters=num_clu
【文件预览】:
kmeans_pytorch-master
----setup.py(2KB)
----batch_processing.ipynb(136KB)
----.gitignore(205B)
----MANIFEST.in(117B)
----LICENSE(1KB)
----cpu_vs_gpu.ipynb(59KB)
----README.md(2KB)
----tests()
--------test_project.py(279B)
----makefile(270B)
----kmeans_pytorch()
--------__init__.py(5KB)
--------main.py(138B)
----example.ipynb(48KB)
----docs()
--------index.md(1KB)
--------chapters()
--------static()
--------_config.yml(829B)
--------_layouts()
--------404.html(260B)