org.maochen.nlp.ml.classifier

Class LabelIndexer

java.lang.Object

org.maochen.nlp.ml.classifier.LabelIndexer

public class LabelIndexer
extends Object

Created by Maochen on 12/4/14.

Field Summary

Fields

Modifier and Type	Field and Description
com.google.common.collect.BiMap<String,Integer>	labelIndexer

Constructor Summary

Constructors

Constructor and Description
LabelIndexer(List<Tuple> trainingData)

Method Summary

Modifier and Type	Method and Description
Map<String,Double>	convertMapKey(Map<Integer,Double> probs)
int	getIndex(String label)
Set<Integer>	getIndexSet()
String	getLabel(int index)
int	getLabelSize()
boolean	hasLabel(String label)
void	putByLabels(List<String> labels)

Methods inherited from class java.lang.Object

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

Field Detail

labelIndexer

public com.google.common.collect.BiMap<String,Integer> labelIndexer

Constructor Detail

LabelIndexer

public LabelIndexer(List<Tuple> trainingData)

Method Detail

getIndex

public int getIndex(String label)

getLabel

public String getLabel(int index)

putByLabels

public void putByLabels(List<String> labels)

hasLabel

public boolean hasLabel(String label)

getIndexSet

public Set<Integer> getIndexSet()

getLabelSize

public int getLabelSize()

convertMapKey

public Map<String,Double> convertMapKey(Map<Integer,Double> probs)