Class OneLabelGEConstraints
- java.lang.Object
-
- cc.mallet.fst.semi_supervised.constraints.OneLabelGEConstraints
-
- All Implemented Interfaces:
GEConstraint
- Direct Known Subclasses:
OneLabelKLGEConstraints,OneLabelL2GEConstraints
public abstract class OneLabelGEConstraints extends java.lang.Object implements GEConstraint
A set of constraints on distributions over single labels conditioned on the presence of input features. Subclasses are to be used with GE. Multiple constraints are grouped together here to make things more efficient.- Author:
- Gregory Druck
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description protected classOneLabelGEConstraints.OneLabelGEConstraint
-
Field Summary
Fields Modifier and Type Field Description protected com.carrotsearch.hppc.IntArrayListcacheprotected com.carrotsearch.hppc.IntObjectHashMap<OneLabelGEConstraints.OneLabelGEConstraint>constraintsprotected StateLabelMapmap
-
Constructor Summary
Constructors Modifier Constructor Description OneLabelGEConstraints()protectedOneLabelGEConstraints(com.carrotsearch.hppc.IntObjectHashMap<OneLabelGEConstraints.OneLabelGEConstraint> constraints, StateLabelMap map)
-
Method Summary
All Methods Instance Methods Abstract Methods Concrete Methods Modifier and Type Method Description abstract voidaddConstraint(int fi, double[] target, double weight)voidcomputeExpectations(java.util.ArrayList<SumLattice> lattices)Compute expectations using cached lattices.doublegetCompositeConstraintFeatureValue(FeatureVector fv, int ip, int si1, int si2)Computes the composite constraint feature value (over all constraint features) for FeatureVector fv and a transition from state li1 to li2.abstract doublegetValue()Returns the total constraint value.booleanisOneStateConstraint()voidpreProcess(FeatureVector fv)Gives the constraint the option to do some caching using only the FeatureVector.java.util.BitSetpreProcess(InstanceList data)voidsetStateLabelMap(StateLabelMap map)Sets that map between the state indices and label indices.voidzeroExpectations()Zero expectation values.-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface cc.mallet.fst.semi_supervised.constraints.GEConstraint
copy
-
-
-
-
Field Detail
-
constraints
protected com.carrotsearch.hppc.IntObjectHashMap<OneLabelGEConstraints.OneLabelGEConstraint> constraints
-
map
protected StateLabelMap map
-
cache
protected com.carrotsearch.hppc.IntArrayList cache
-
-
Constructor Detail
-
OneLabelGEConstraints
public OneLabelGEConstraints()
-
OneLabelGEConstraints
protected OneLabelGEConstraints(com.carrotsearch.hppc.IntObjectHashMap<OneLabelGEConstraints.OneLabelGEConstraint> constraints, StateLabelMap map)
-
-
Method Detail
-
addConstraint
public abstract void addConstraint(int fi, double[] target, double weight)
-
isOneStateConstraint
public boolean isOneStateConstraint()
- Specified by:
isOneStateConstraintin interfaceGEConstraint- Returns:
- true if constraint feature only considers one state
-
setStateLabelMap
public void setStateLabelMap(StateLabelMap map)
Description copied from interface:GEConstraintSets that map between the state indices and label indices.- Specified by:
setStateLabelMapin interfaceGEConstraint- Parameters:
map- StateLabelMap
-
preProcess
public void preProcess(FeatureVector fv)
Description copied from interface:GEConstraintGives the constraint the option to do some caching using only the FeatureVector. For example, the constrained input features could be cached.- Specified by:
preProcessin interfaceGEConstraint- Parameters:
fv- FeatureVector input
-
preProcess
public java.util.BitSet preProcess(InstanceList data)
- Specified by:
preProcessin interfaceGEConstraint- Parameters:
data- Unlabeled data- Returns:
- Returns a bitset of the size of the data, with the bit set if a constraint feature fires in that instance.
-
getCompositeConstraintFeatureValue
public double getCompositeConstraintFeatureValue(FeatureVector fv, int ip, int si1, int si2)
Description copied from interface:GEConstraintComputes the composite constraint feature value (over all constraint features) for FeatureVector fv and a transition from state li1 to li2.- Specified by:
getCompositeConstraintFeatureValuein interfaceGEConstraint- Parameters:
fv- FeatureVector on transitionip- Position of input in sequencesi1- Source state index for transitionsi2- Destination state index for transition- Returns:
- Constraint feature value
-
getValue
public abstract double getValue()
Description copied from interface:GEConstraintReturns the total constraint value.- Specified by:
getValuein interfaceGEConstraint- Returns:
- Constraint value
-
zeroExpectations
public void zeroExpectations()
Description copied from interface:GEConstraintZero expectation values. Called before re-computing gradient.- Specified by:
zeroExpectationsin interfaceGEConstraint
-
computeExpectations
public void computeExpectations(java.util.ArrayList<SumLattice> lattices)
Description copied from interface:GEConstraintCompute expectations using cached lattices.- Specified by:
computeExpectationsin interfaceGEConstraint- Parameters:
lattices- Cached SumLattices
-
-