Reduced the bracket size for IWHB tests.
*/
package org.linqs.psl.application.learning.weight.search;
+import org.linqs.psl.application.learning.weight.VotedPerceptron;
import org.linqs.psl.application.learning.weight.WeightLearningApplication;
import org.linqs.psl.application.learning.weight.WeightLearningTest;
import org.linqs.psl.config.Config;
@@ -33,6 +34,8 @@ public class InitialWeightHyperbandTest extends WeightLearningTest {
@Override
protected WeightLearningApplication getWLA() {
Config.setProperty(Hyperband.NUM_BRACKETS_KEY, 1);
+ Config.setProperty(Hyperband.BASE_BRACKET_SIZE_KEY, 2);
+ Config.setProperty(VotedPerceptron.NUM_STEPS_KEY, 5);
return new InitialWeightHyperband(info.model.getRules(), weightLearningTrainDB, weightLearningTruthDB);
}
