diff --git a/clustering/KMeans.java b/clustering/KMeans.java new file mode 100644 index 0000000..69d8f17 --- /dev/null +++ b/clustering/KMeans.java @@ -0,0 +1,112 @@ +package clustering; +import java.util.ArrayList; +import java.util.List; +import helperClasses.DataPoint; +import helperClasses.Serialization; + +public class KMeans { + static int K = 3; + public List means = new ArrayList(); + public List> clusters = new ArrayList>(); + + public void importClusterData(){ + ArrayList> c = null; + Serialization S = new Serialization(); + clusters = S.clustDeserialize(c, "kMeansClusterData.ser"); + } + + public double getClusterMean(int current){ + double sum = 0, mean=clusters.get(current).get(0).x; + for(int i=1; i> reviseClusters(){ + for(int i=0; i()); + } + for(int i=0; i> c = null; + Serialization S = new Serialization(); + c = S.clustDeserialize(c, "kMeansClusterData.ser"); + System.out.println("Clusters : "); + for(int i=0; i