org.apache.ignite.ml.knn.ann

Class ANNClassificationModel

java.lang.Object

org.apache.ignite.ml.knn.NNClassificationModel

org.apache.ignite.ml.knn.ann.ANNClassificationModel

All Implemented Interfaces:

Serializable, Function<Vector,Double>, Exportable<KNNModelFormat>, IgniteFunction<Vector,Double>, Model<Vector,Double>

public class ANNClassificationModel
extends NNClassificationModel

ANN model to predict labels in multi-class classification task.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.ignite.ml.knn.NNClassificationModel

distanceMeasure, k, stgy

Constructor Summary

Constructors

Constructor and Description
ANNClassificationModel(LabeledVectorSet<ProbableLabel,LabeledVector> centers, ANNClassificationTrainer.CentroidStat centroindsStat) Build the model based on a candidates set.

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector v)
boolean	equals(Object obj)
LabeledVectorSet<ProbableLabel,LabeledVector>	getCandidates()
ANNClassificationTrainer.CentroidStat	getCentroindsStat()
int	hashCode()
<P> void	saveModel(Exporter<KNNModelFormat,P> exporter, P path) Save model by the given path.
String	toString()
String	toString(boolean pretty)

Methods inherited from class org.apache.ignite.ml.knn.NNClassificationModel

buildLabeledDatasetOnListOfVectors, copyParametersFrom, getClassVoteForVector, getClassWithMaxVotes, getDistanceMeasure, getDistances, getKClosestVectors, putDistanceIdxPair, withDistanceMeasure, withK, withStrategy

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.apache.ignite.ml.Model

combine

Methods inherited from interface java.util.function.Function

andThen, compose, identity

Constructor Detail

ANNClassificationModel

public ANNClassificationModel(LabeledVectorSet<ProbableLabel,LabeledVector> centers,
                              ANNClassificationTrainer.CentroidStat centroindsStat)

Build the model based on a candidates set.

Parameters:

centers - The candidates set.

centroindsStat - The stat about centroids.

Method Detail

getCandidates

public LabeledVectorSet<ProbableLabel,LabeledVector> getCandidates()

getCentroindsStat

public ANNClassificationTrainer.CentroidStat getCentroindsStat()

apply

public Double apply(Vector v)

saveModel

public <P> void saveModel(Exporter<KNNModelFormat,P> exporter,
                          P path)

Description copied from interface: Exportable

Save model by the given path.

Specified by:

saveModel in interface Exportable<KNNModelFormat>

Specified by:

saveModel in class NNClassificationModel

Parameters:

exporter - Exporter.

path - Path to saved model.

hashCode

public int hashCode()

Overrides:

hashCode in class NNClassificationModel

equals

public boolean equals(Object obj)

Overrides:

equals in class NNClassificationModel

toString

public String toString()

Overrides:

toString in class NNClassificationModel

toString

public String toString(boolean pretty)

Specified by:

toString in interface Model<Vector,Double>

Overrides:

toString in class NNClassificationModel

Parameters:

pretty - Use pretty mode.