Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added hard and soft Kmeans clustering with tests #71

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

kenluck2001
Copy link
Contributor

@kenluck2001 kenluck2001 commented May 27, 2021

This submission addresses the issue tracked in #69
We have implemented a soft and hard version of kmeans clustering.
The works done can be summarized as follows:

  1. Hard kmeans clustering with fixed assignment of data points to only one cluster at a time.
  2. Soft kmeans clustering with probabilistic assignment of data points. Each data point has a membership degree in each cluster. The highest probable cluster could then be assigned as the cluster index of the data. Alternatively, the probability distribution can be used for any other purpose as it captures our uncertainty of the clustering routine.
  • Is the code you are submitting your own work? Yes it is my original work
  • Did you properly attribute the authors of any code you referenced? Yes I did
  • Did you write unit tests for your new model? Yes I added unit tests by comparing algorithms to base implementation in scikit-learn.
  • Does your submission pass the unit tests? Yes it does
  • Did you write documentation for your new model? For now, it is readme and code documentation
  • Have you formatted your code using the black deaults? Only followed the numpy styling guide

@kenluck2001 kenluck2001 changed the title Added hard and soft Kmeans clustering with tests tests Added hard and soft Kmeans clustering with tests May 27, 2021
@ddbourgin
Copy link
Owner

Thanks for this, @kenluck2001 ! At first glance this looks great - I'm going to reserve some time to go through this more thoroughly shortly.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants