44
An Overview of the scikit-learn Clustering Package
Clustering is an unsupervised Machine Learning technique, where there is neither a training set nor predefined classes. Clustering is used when there are many records, which should be grouped according to similarity criteria, such as distance.
A clustering algorithm takes a dataset as input and returns a list of labels as output, corresponding to the associated clusters.
Cluster analysis is an iterative process where, at each step, the current iteration is evaluated and used to feedback into changes to the algorithm in the next iteration, until the desired result is obtained.
The scikit-learn library provides a subpackage, called sklearn.cluster, which provides the most common clustering algorithms.
In this article, I describe:
- class and functions provided by sklearn.cluster tuning parameters
- evaluation metrics for clustering algorithms
The sklearn.cluster subpackage defines two ways to apply a clustering algorithm: classes and functions.
In the class strategy, you should create an instance of the desired clustering class algorithm, by also specifying the class parameters. Then you fit the algorithm with data and, finally, you can use the fitted model to predict clusters:
from sklearn.cluster import AffinityPropagation
model = AffinityPropagation()
model.fit(X)
labels = model.predict(X)
In addition to the class definition, Scikit-learn provides functions to perform the model fitting. With respect to classes, functions can be used when there is a single dataset, which must be analyzed just once, in a single spot.
In this case, it is sufficient to call the function, in order to get clusterized data:
from sklearn.cluster import affinity_propagation
result = affinity_propagatiom(X)
Clustering algorithms can be split into two big families, depending on the main parameter to be tuned:
the number of clusters to discover in the data
minimum distance between observations.
Usually, in this group of clustering algorithms, you should tune at least the maximum number of clusters to find. In scikit-learn, often this parameter is called n_clusters.
The sklearn.cluster package provides the following clustering algorithms belonging to this category (both the (both the class and the function is shown for each provided algorithm):
Continue Reading on Towards Data Science
44