ai.libs.jaicore.search.algorithms.mdp.mcts.uuct.UUCTFactory Maven / Gradle / Ivy
package ai.libs.jaicore.search.algorithms.mdp.mcts.uuct;
import ai.libs.jaicore.search.algorithms.mdp.mcts.MCTS;
import ai.libs.jaicore.search.algorithms.mdp.mcts.MCTSFactory;
import ai.libs.jaicore.search.probleminputs.IMDP;
public class UUCTFactory extends MCTSFactory> {
private IUCBUtilityFunction utility;
public IUCBUtilityFunction getUtility() {
return this.utility;
}
public void setUtility(final IUCBUtilityFunction utility) {
this.utility = utility;
}
@Override
public MCTS getAlgorithm(final IMDP input) {
return new UUCT<>(input, this.getDefaultPolicy(true), this.utility, this.getMaxIterations(), this.getGamma(), this.getEpsilon(), this.isTabooExhaustedNodes());
}
}