search

pycaret / pycaret

An open-source, low-code machine learning library in Python
https://www.pycaret.org
MIT License
8.93k stars 1.77k forks source link

Optimal Number of Clusters #1504

Open deepakmehta79 opened 3 years ago

deepakmehta79 commented 3 years ago

Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is.

How to extract an optimal number of clusters using methods like elbow, silhouette.

Describe the solution you'd like

tune_model api could perhaps allow to compute the optimal number of features

Describe alternatives you've considered

Currently using affinity propagation to estimate the number of clusters

moezali1 commented 3 years ago

@deepakmehta79 I wouldn't mind that addition in the existing tune_model function of pycaret.clustering. Would you like to work on this enhancement?