How to set sample_weights in fit_predict

jokofa / torch_kmeans

PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.

MIT License

57 stars 6 forks source link

How to set sample_weights in fit_predict #10

Open yenanjing opened 7 months ago

yenanjing commented 7 months ago

This is a very useful tool, but I think something like the sample_weights parameter on sklearn, where the side doesn't seem to be implemented, is there a trick to use or future or increase?