package image.processing.algorithms.clustering.kmeans;
/* This file is copyright (c) 2008-2013 Philippe Fournier-Viger
*
* This file is part of the SPMF DATA MINING SOFTWARE
* (http://www.philippe-fournier-viger.com/spmf).
*
* SPMF is free software: you can redistribute it and/or modify it under the
* terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* SPMF is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
* You should have received a copy of the GNU General Public License along with
* SPMF. If not, see <http://www.gnu.org/licenses/>.
*/
import image.processing.ImageManipulation;
import image.processing.ImageSorter;
import image.processing.Interface;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import image.processing.patterns.cluster.Cluster;
import image.processing.patterns.cluster.DoubleArray;
import image.processing.tools.MemoryLogger;
/**
* An implementation of the K-means algorithm (J. MacQueen, 1967).
* <br/><br/>
*
* The K-means algorithm steps are (text from Wikipedia) : 1) Choose the number of clusters, k.
* * 2) Randomly generate k clusters and determine the cluster centers, or directly
* generate k random points as cluster centers. 3) Assign each point to the
* nearest cluster center. 4) Recompute the new cluster centers. 5) Repeat the two
* previous steps until some convergence criterion is met (usually that the
* assignment hasn't changed).
*
* @author Philippe Fournier-Viger
*/
public class AlgoKMeansCol {
// The list of clusters generated
private List<Cluster> clusters = null;
// A random number generator because K-Means is a randomized algorithm
private final static Random random = new Random(System.currentTimeMillis());
// For statistics
private long startTimestamp; // the start time of the latest execution
private long endTimestamp; // the end time of the latest execution
private long iterationCount; // the number of iterations that was performed
private int vectorCount;
/**
* Default constructor
*/
public AlgoKMeansCol() {
}
/**
* Run the K-Means algorithm
* @param inputFile an input file path containing a list of vectors of double values
* @param k the parameter k
* @return a list of clusters (some of them may be empty)
* @throws IOException exception if an error while writing the file occurs
*/
public List<Cluster> runAlgorithm(String inputFile, int k) throws NumberFormatException, IOException {
// record the start time
startTimestamp = System.currentTimeMillis();
// reset the number of iterations
iterationCount =0;
//vectorCount
vectorCount = 0;
// Structure to store the vectors from the file
List<DoubleArray> vectors = new ArrayList<DoubleArray>();
// variables to store the minimum and maximum values in vectors
double minValue = Integer.MAX_VALUE;
double maxValue = 0;
// read the vectors from the input file
BufferedReader reader = new BufferedReader(new FileReader(inputFile));
String line;
// for each line until the end of the file
while (((line = reader.readLine()) != null)) {
// if the line is a comment, is empty or is a
// kind of metadata
if (line.isEmpty() == true ||
line.charAt(0) == '#' || line.charAt(0) == '%'
|| line.charAt(0) == '@') {
continue;
}
// split the line by spaces
String[] lineSplited = line.split(" ");
// create a vector of double
double [] vector = new double[lineSplited.length];
// for each value of the current line
for (int i=0; i< lineSplited.length; i++) {
// convert to double
double value = Double.parseDouble(lineSplited[i]);
// check if it is the min or max
if(value < minValue){
minValue = value;
}
if(value > maxValue){
maxValue = value;
}
// add the value to the current vector
vector[i] = value;
}
// add the vector to the list of vectors
vectors.add(new DoubleArray(vector, vectorCount));
vectorCount++;
}
// close the file
reader.close();
// Create a list of clusters
clusters = new ArrayList<Cluster>();
// Get the size of vectors
int vectorsSize = vectors.get(0).data.length;
System.out.println("Vector size: " + vectors.size());
// SPECIAL CASE: If only one vector
if (vectors.size() == 1) {
// Create a single cluster and return it
DoubleArray vector = vectors.get(0);
Cluster cluster = new Cluster(vectorsSize);
cluster.addVector(vector);
clusters.add(cluster);
return clusters;
}
// (1) Randomly generate k empty clusters with a random mean (cluster
// center)
for(int i=0; i< k; i++){
int rand = 0 + (int)(Math.random() * vectors.size()-1);
DoubleArray meanVector = vectors.get(rand);
Cluster cluster = new Cluster(vectorsSize);
cluster.setMean(meanVector);
clusters.add(cluster);
}
// (2) Repeat the two next steps until the assignment hasn't changed
boolean changed;
while(true) {
iterationCount++;
changed = false;
// (2.1) Assign each point to the nearest cluster center.
// / for each vector
for (DoubleArray vector : vectors) {
// find the nearest cluster and the cluster containing the item
Cluster nearestCluster = null;
Cluster containingCluster = null;
double distanceToNearestCluster = Double.MAX_VALUE;
// for each cluster
for (Cluster cluster : clusters) {
// calculate the distance of the cluster mean to the vector
double distance = euclideanDistance(cluster.getmean(), vector);
// if it is the smallest distance until now, record this cluster
// and the distance
if (distance < distanceToNearestCluster) {
nearestCluster = cluster;
distanceToNearestCluster = distance;
}
// if the cluster contain the vector already,
// remember that too!
if (cluster.contains(vector)) {
containingCluster = cluster;
}
}
// if the nearest cluster is not the cluster containing
// the vector
if (containingCluster != nearestCluster) {
// remove the vector from the containing cluster
if (containingCluster != null) {
containingCluster.remove(vector);
}
// add the vector to the nearest cluster
nearestCluster.addVector(vector);
changed = true;
}
}
// check the memory usage
MemoryLogger.getInstance().checkMemory();
if(!changed){ // exit condition for main loop
break;
}
// (2.2) Recompute the new cluster means
for (Cluster cluster : clusters) {
cluster.recomputeClusterMean();
}
}
// check memory usage
MemoryLogger.getInstance().checkMemory();
// record end time
endTimestamp = System.currentTimeMillis();
// return the clusters
return clusters;
}
/**
* Generate a random vector.
* @param minValue the minimum value allowed
* @param maxValue the maximum value allowed
* @param vectorsSize the desired vector size
* @return the random vector
*/
private DoubleArray generateRandomVector(double minValue, double maxValue,
int vectorsSize) {
// create a new vector
double[] vector = new double[vectorsSize];
// for each position generate a random number
for(int i=0; i < vectorsSize; i++){
vector[i] = (random.nextDouble() * (maxValue - minValue)) + minValue;
}
// return the vector
return new DoubleArray(vector, 0);
}
/**
* Calculate the eucledian distance between two vectors of doubles.
* @param vector1 the first vector
* @param vector2 the second vector
* @return the distance
*/
private double euclideanDistance(DoubleArray vector1, DoubleArray vector2) {
double sum =0;
for(int i=0; i< vector1.data.length; i++){
sum += Math.pow(vector1.data[i] - vector2.data[i], 2);
}
return Math.sqrt(sum);
}
/**
* Print statistics of the latest execution to System.out.
*/
public void printStatistics() {
System.out.println("========== KMEANS - STATS ============");
System.out.println(" Total time ~: " + (endTimestamp - startTimestamp)
+ " ms");
System.out.println(" Max memory:" + MemoryLogger.getInstance().getMaxMemory() + " mb ");
System.out.println(" Iteration count: " + iterationCount);
System.out.println("=====================================");
}
/**
* Save the clusters to an output file
* @param output the output file path
* @throws IOException exception if there is some writing error.
*/
public void saveToFile(String output) throws IOException {
BufferedWriter writer = new BufferedWriter(new FileWriter(output));
int[] tempRow;
ImageSorter.colClusterOrder = new ArrayList<int[]>();
//Interface.colourClusterArea.setText(null);
//Interface.colourClusterArea.append("Colour cluster amount: " + (clusters.size()) + "\n\n");
ImageManipulation.initColResultsImages();
// for each cluster
for(int i=0; i< clusters.size(); i++){
//initialise tempRow
tempRow = new int[clusters.get(i).getVectors().size()];
// if the cluster is not empty
if(clusters.get(i).getVectors().size() >= 1){
// write the cluster
writer.write("Cluster"+ i + ": " + clusters.get(i).toString());
//Interface.colourClusterArea.append("Cluster " + i + ":\n");
for(int j = 0; j < clusters.get(i).getVectors().size(); j++)
{
//Interface.colourClusterArea.append(" -contains "
// + ImageSorter.photoSet[clusters.get(i).getVectors().get(j).getNum()].getShortName()
// + "\n");
ImageManipulation.updateColResultsImages(clusters.get(i).getVectors().get(j).getNum(), i);
//Add current image number to temp array
tempRow[j] = clusters.get(i).getVectors().get(j).getNum();
}
// if not the last cluster, add a line return
if(i < clusters.size()-1){
writer.newLine();
}
}
else{
//Interface.colourClusterArea.append("Cluster " + i + ": EMPTY \n");
}
//Add new vector order for cluster to reference array
ImageSorter.colClusterOrder.add(tempRow);
}
ImageManipulation.finaliseColResultsImages();
// close the file
writer.close();
}
}