use of edu.cmu.ml.proppr.util.math.SimpleParamVector in project ProPPR by TeamCohen.
the class SRWTest method moreSetup.
@Override
public void moreSetup(LearningGraphBuilder lgb) {
uniformParams = srw.getRegularizer().setupParams(new SimpleParamVector<String>(new ConcurrentHashMap<String, Double>()));
for (String n : new String[] { "fromb", "tob", "fromr", "tor" }) uniformParams.put(n, srw.getSquashingFunction().defaultValue());
startVec = new TIntDoubleHashMap();
startVec.put(nodes.getId("r0"), 1.0);
}
Aggregations