Java IntIter类代码示例

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
/**
 * Gets the marginal distribution over a subset of the variables in this
 * factor, optionally normalized.
 * 
 * @param vars The subset of variables for the marginal distribution. This will sum over all variables not in this set.
 * @param normalize Whether to normalize the resulting distribution.
 * @return The marginal distribution.
 */
public VarTensor getMarginal(VarSet vars, boolean normalize) {
    VarSet margVars = new VarSet(this.vars);
    margVars.retainAll(vars);
    
    VarTensor marg = new VarTensor(s, margVars, s.zero());
    if (margVars.size() == 0) {
        return marg;
    }
    
    IntIter iter = margVars.getConfigIter(this.vars);
    for (int i=0; i<this.values.length; i++) {
        int j = iter.next();
        marg.values[j] = s.plus(marg.values[j], this.values[i]);
    }
    
    if (normalize) {
        marg.normalize();
    }
    
    return marg;
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
public VarTensor getClamped(VarConfig clmpVarConfig) {
    if (clmpVarConfig.size() == 0) {
        return new VarTensor(this);
    }
    VarSet clmpVars = clmpVarConfig.getVars();
    VarSet unclmpVars = new VarSet(this.vars);
    unclmpVars.removeAll(clmpVars); 

    VarTensor clmp = new VarTensor(s, unclmpVars);
    IntIter iter = IndexForVc.getConfigIter(this.vars, clmpVarConfig);
    
    if (clmp.values.length > 0) {
        for (int c=0; c<clmp.values.length; c++) {
            int config = iter.next();
            clmp.values[c] = this.values[config];
        }
    }
    return clmp;
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
/**
 * Returns a list of pairs (i,j) for which a role variable should be built
 * according to the provided roleStructure.
 *
 * @param isent
 *            Sentence to get pairs for.
 * @param rS
 *            RoleStructure describing which pairs to include.
 * @param allowPredArgSelfLoops
 *            boolean indicating if pairs (i,i) should be included.
 * @return
 */
public static List<Pair<Integer, Integer>> getPossibleRolePairs(int n, IntSet knownPreds, Collection<Pair<Integer, Integer>> knownPairs, Collection<Pair<Integer, Integer>> pairsToSkip, RoleStructure rS,
        boolean allowPredArgSelfLoops) {
    List<Pair<Integer, Integer>> toReturn = new ArrayList<>();
    if (rS == RoleStructure.PAIRS_GIVEN) {
        toReturn.addAll(knownPairs);
        toReturn.sort(ComparablePair.naturalOrder());
    } else if (rS == RoleStructure.PREDS_GIVEN) {
        // CoNLL-friendly model; preds given
        IntIter iter = knownPreds.iterator();
        while (iter.hasNext()) {
            int i = iter.next();
            for (int j = 0; j < n; j++) {
                if (i == j && !allowPredArgSelfLoops) {
                    continue;
                }
                toReturn.add(new Pair<>(i, j));
            }
        }
    } else if (rS == RoleStructure.ALL_PAIRS) {
        // n**2 model
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (i != j || allowPredArgSelfLoops) {
                    toReturn.add(new Pair<>(i, j));
                }
            }
        }
    } else if (rS == RoleStructure.NO_ROLES) {
        // No role variables.
    } else {
        throw new IllegalArgumentException("Unsupported model structure: " + rS);
    }
    if (pairsToSkip != null) {
        toReturn.removeAll(pairsToSkip);
    }
    return toReturn;
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
/**
 * Gets an array version of the configuration iterator.
 * @see edu.jhu.pacaya.gm.model.VarSet#getConfigIter
 */
public int[] getConfigArr(VarSet vars) {        
    IntArrayList a = new IntArrayList(vars.calcNumConfigs());
    IntIter iter = getConfigIter(vars);
    while (iter.hasNext()) {
        a.add(iter.next());
    }
    return a.toNativeArray();
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
/**
 * Gets an array version of the configuration iterator.
 * @see edu.jhu.pacaya.gm.model.IndexForVc#getConfigIter
 */
public static int[] getConfigArr(VarSet vars, VarConfig config) {        
    IntArrayList a = new IntArrayList(config.getVars().calcNumConfigs());
    IntIter iter = getConfigIter(vars, config);
    while (iter.hasNext()) {
        a.add(iter.next());
    }
    return a.toNativeArray();
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
public IntIter iterator() {
    return new IntArrayIter(map.getIndices());
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
public IntIter indices() {
    return new SparseIdxIter(idx, top);
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
/**
 * Gets an iterator over the configurations of the variables for this
 * variable set, such that the c'th configuration in the iterator will
 * correspond to the variable setting where all the values of the
 * variables in this set take on the same values as the c'th
 * configuration of the variables in vars, and all the other variables
 * take on the zero state.
 * 
 * This is a trick from libDAI for efficiently computing factor
 * products.
 * 
 * @param vars The variable set to which the iterator should be aligned.
 * @return The iterator.
 */
public IntIter getConfigIter(VarSet vars) {
    return new IndexFor(this, vars);
}
 

import edu.jhu.prim.iter.IntIter; //导入依赖的package包/类
IntIter iterator();