search

subhadarship / kmeans_pytorch

kmeans using PyTorch
https://subhadarship.github.io/kmeans_pytorch
MIT License
479 stars 77 forks source link

Initialize with setting Seed #30

Closed sprenkamp closed 2 years ago

sprenkamp commented 2 years ago

Hi, as far as I understand it is not possible to set a seed to reproduce the results of the algorithm. I am I correct here? I looked at the code and it would be quite simple to edit I believe, are you interested in adding the function and can I participate?

subhadarship commented 2 years ago

Yes the current implementation doesn't set a random seed. Totally feel free to contribute and create a pull request

sprenkamp commented 2 years ago

Hi subhadarship, can you give me Permission to push to the repo? I have a first draft that works, at which you can look at.

subhadarship commented 2 years ago

Create a pull request