ai.libs.jaicore.search.algorithms.mdp.mcts.ensemble.EnsembleMCTSFactory Maven / Gradle / Ivy
package ai.libs.jaicore.search.algorithms.mdp.mcts.ensemble;
import java.util.ArrayList;
import java.util.Collection;
import ai.libs.jaicore.search.algorithms.mdp.mcts.IPathUpdatablePolicy;
import ai.libs.jaicore.search.algorithms.mdp.mcts.MCTSFactory;
import ai.libs.jaicore.search.probleminputs.IMDP;
public class EnsembleMCTSFactory extends MCTSFactory> {
private Collection> treePolicies = new ArrayList<>();
public Collection> getTreePolicies() {
return this.treePolicies;
}
public void setTreePolicies(final Collection> treePolicies) {
this.treePolicies = treePolicies;
}
@Override
public EnsembleMCTS getAlgorithm(final IMDP input) {
return new EnsembleMCTS<>(input, this.getDefaultPolicy(true), this.treePolicies, this.getMaxIterations(), this.getGamma(), this.getEpsilon(), this.isTabooExhaustedNodes());
}
}