Bug fixes (#10, #16), adding soft dynamic time warping measure

subhadarship / kmeans_pytorch

kmeans using PyTorch

https://subhadarship.github.io/kmeans_pytorch

MIT License

479 stars 77 forks source link

Closed zaemyung closed 3 years ago

zaemyung commented 3 years ago

Adds @DeepOzean 's suggestion on replacing an empty cluster with a random one. #16
Fixes the issue where the device is always set to CPU when calculating distances. #10
Adds @Maghoumi implementation of soft dynamic time warping measure (soft-dtw) #20
- https://github.com/Maghoumi/pytorch-softdtw-cuda