Evaluating a clustering model

Now, we'll look at how to evaluate a clustering model that has been trained. Let's look at the code and see how this is done.

We'll be using the following classes:

import weka.core.Instances;import weka.core.converters.ConverterUtils.DataSource;import weka.clusterers.SimpleKMeans;import weka.clusterers.ClusterEvaluation;

We'll use the ClusterEvaluation class from the weka.clusterers package for evaluation.

First, we will read our dataset into our DataSource object and assign it to the Instances object. Then, we'll create our k-means object and specify the number of clusters that we want to create. Next, we will train our clustering algorithm using the buildClusterer method; then, we'll print it using println ...

Get Hands-On Artificial Intelligence with Java for Beginners now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.