Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 812 Bytes

README.md

File metadata and controls

32 lines (23 loc) · 812 Bytes
Raw

fuzzy-c-means

Fuzzy c-means Clustering

Description

This implementation is based on the paper FCM: The fuzzy c-means clustering algorithm by: James C.Bezdek, Robert Ehrlich, and William Full

To run the tests

sh run_tests.sh

To run the coverage

sh run_coverage.sh

Install via pip

pip install fuzzycmeans

How to use it

  1. Fit the model. This is to cluster any given data X.
X = np.array([[1, 1], [1, 2], [2, 2], [0, 0], [0, 0]])
fcm = FCM(n_clusters=3, max_iter=1)
fcm.fit(X, [0, 0, 0, 1, 2])
  1. (Optional.) Use the model to assign new data points to existing clusters. Note that the predict function would return the membership as this a fuzzy clustering.
Y = np.array([[1, 2], [2, 2], [3, 1], [2, 1], [6, 8]])
membership = fcm.predict(Y)