Package ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling

Class KMeansStratiAssigner

java.lang.Object

ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling.ClusterStratiAssigner

ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling.KMeansStratiAssigner

All Implemented Interfaces:

IStratiAssigner, org.api4.java.common.control.IParallelizable

public class KMeansStratiAssigner
extends ClusterStratiAssigner

Cluster the data set with k-means into k Clusters, where each cluster stands for one stratum. The datapoint assignment is performed with a lookup in the clusters.

Field Summary

Fields inherited from class ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling.ClusterStratiAssigner

distanceMeasure, randomSeed

Constructor Summary

Constructors

Constructor	Description
KMeansStratiAssigner(org.apache.commons.math3.ml.distance.DistanceMeasure distanceMeasure, int randomSeed)	Constructor for KMeansStratiAssigner.

Method Summary

Modifier and Type	Method	Description
void	init(org.api4.java.ai.ml.core.dataset.IDataset<?> dataset, int stratiAmount)	Initialize custom assigner if necessary.

Methods inherited from class ai.libs.jaicore.ml.core.filter.sampling.inmemory.stratified.sampling.ClusterStratiAssigner

assignToStrati, getClusters, getNumCPUs, setClusters, setDataset, setNumCPUs

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KMeansStratiAssigner

public KMeansStratiAssigner(org.apache.commons.math3.ml.distance.DistanceMeasure distanceMeasure,
                            int randomSeed)

Constructor for KMeansStratiAssigner.

Parameters:

distanceMeasure - Distance measure for datapoints, for example Manhattan or Euclidian.

randomSeed - Seed for random numbers.

Method Detail

init

public void init(org.api4.java.ai.ml.core.dataset.IDataset<?> dataset,
                 int stratiAmount)

Description copied from interface: IStratiAssigner

Initialize custom assigner if necessary.

Parameters:

dataset - The dataset the datapoints will be sampled from.

stratiAmount - The predetermined amount of strati the dataset will be stratified into.