Segmenting Audience with KMeans and Voronoi Diagram using Spark and MLlib
2014/07/11 2 Comments
Analyzing huge data set to extract meaningful properties can be a difficult task. Several methods have been developed for the last 50 years to find hidden information.
Clustering algorithms can be used to group similar news like in Google News, find areas with high crime concentration, find trends, .. and segment the data into groups. This segmentation can be used for instance by publisher to reach a specific target audience.
In this post, we will be using the k-means clustering algorithm implemented in Spark Machine Learning Library(MLLib) to segment the dataset by geolocation .
The k-mean clustering algorithm is an unsupervised algorithm meaning that you don’t need to provide a training example for it to work(unlike neural network, SVM, Naives Bayes classifiers, …). It partitions observations into clusters in which each observation belongs to the cluster with the nearest mean. The algorithm takes as input the observations, the number of clusters(denoted k) that we want to partition the observation into and the number of iterations. It gives as a result the centers of the clusters.
The algorithm works as follow:
- Take k random observations out of the dataset. Set the k centers of the clusters to those points
- For each observation, find the cluster center which is the closest and assign this observation to this cluster
- For each cluster, compute the new center by taking the average of the features of the observations assigned to this dataset
- Go back to 2 and repeat this for a given number of iterations
The centers of the clusters will converge and will minimize the cost function which is the sum of the square distance of each observation to their assigned cluster centers.
This minimum might be a local optimum and will depend on the observation that were randomly taken at the beginning of the algorithm.
Read more of this post
Recent Comments