org.apache.spark.ml.classification

Class LabelConverter

java.lang.Object

org.apache.spark.ml.classification.LabelConverter

public class LabelConverter
extends java.lang.Object

Label to vector converter.

Constructor Summary

Constructors

Constructor and Description
LabelConverter()

Method Summary

Methods

Modifier and Type	Method and Description
static double	decodeLabel(Vector output) Converts a vector to a label.
static scala.Tuple2<Vector,Vector>	encodeLabeledPoint(LabeledPoint labeledPoint, int labelCount) Encodes a label as a vector.

Methods inherited from class java.lang.Object

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

Constructor Detail

LabelConverter

public LabelConverter()

Method Detail

encodeLabeledPoint

public static scala.Tuple2<Vector,Vector> encodeLabeledPoint(LabeledPoint labeledPoint,
                                             int labelCount)

Encodes a label as a vector. Returns a vector of given length with zeroes at all positions and value 1.0 at the position that corresponds to the label.

Parameters:

labeledPoint - labeled point

labelCount - total number of labels

Returns:

pair of features and vector encoding of a label

decodeLabel

public static double decodeLabel(Vector output)

Converts a vector to a label. Returns the position of the maximal element of a vector.

Parameters:

output - label encoded with a vector

Returns:

label