org.apache.spark.ml.classification

Class LabelConverter

Object

org.apache.spark.ml.classification.LabelConverter

public class LabelConverter
extends Object

Label to vector converter.

Constructor Summary

Constructors

Constructor and Description
LabelConverter()

Method Summary

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 Object

equals, 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