use of org.optaplanner.core.config.solver.SolverConfig in project kogito-apps by kiegroup.
the class ComplexEligibilityDmnCounterfactualExplainerTest method testDMNInvalidCounterfactualExplanation.
@Test
void testDMNInvalidCounterfactualExplanation() throws ExecutionException, InterruptedException, TimeoutException {
PredictionProvider model = getModel();
final List<Output> goal = generateGoal(true, true, 0.6);
List<Feature> features = new LinkedList<>();
// DMN model does not allow loans for age >= 60, so no CF will be possible
features.add(FeatureFactory.newNumericalFeature("age", 61));
features.add(FeatureFactory.newBooleanFeature("hasReferral", true));
features.add(FeatureFactory.newNumericalFeature("monthlySalary", 500, NumericalFeatureDomain.create(10, 10_000)));
final TerminationConfig terminationConfig = new TerminationConfig().withScoreCalculationCountLimit(10_000L);
// for the purpose of this test, only a few steps are necessary
final SolverConfig solverConfig = SolverConfigBuilder.builder().withTerminationConfig(terminationConfig).build();
solverConfig.setRandomSeed((long) 23);
solverConfig.setEnvironmentMode(EnvironmentMode.REPRODUCIBLE);
final CounterfactualConfig counterfactualConfig = new CounterfactualConfig().withSolverConfig(solverConfig).withGoalThreshold(0.01);
final CounterfactualExplainer counterfactualExplainer = new CounterfactualExplainer(counterfactualConfig);
PredictionInput input = new PredictionInput(features);
PredictionOutput output = new PredictionOutput(goal);
Prediction prediction = new CounterfactualPrediction(input, output, null, UUID.randomUUID(), 60L);
final CounterfactualResult counterfactualResult = counterfactualExplainer.explainAsync(prediction, model).get(Config.INSTANCE.getAsyncTimeout(), Config.INSTANCE.getAsyncTimeUnit());
assertFalse(counterfactualResult.isValid());
}
use of org.optaplanner.core.config.solver.SolverConfig in project kogito-apps by kiegroup.
the class ComplexEligibilityDmnCounterfactualExplainerTest method testDMNValidCounterfactualExplanation.
@Test
void testDMNValidCounterfactualExplanation() throws ExecutionException, InterruptedException, TimeoutException {
PredictionProvider model = getModel();
final List<Output> goal = generateGoal(true, true, 0.6);
List<Feature> features = new LinkedList<>();
features.add(FeatureFactory.newNumericalFeature("age", 40));
features.add(FeatureFactory.newBooleanFeature("hasReferral", true));
features.add(FeatureFactory.newNumericalFeature("monthlySalary", 500, NumericalFeatureDomain.create(10, 10_000)));
final TerminationConfig terminationConfig = new TerminationConfig().withScoreCalculationCountLimit(10_000L);
// for the purpose of this test, only a few steps are necessary
final SolverConfig solverConfig = SolverConfigBuilder.builder().withTerminationConfig(terminationConfig).build();
solverConfig.setRandomSeed((long) 23);
solverConfig.setEnvironmentMode(EnvironmentMode.REPRODUCIBLE);
final CounterfactualConfig counterfactualConfig = new CounterfactualConfig().withSolverConfig(solverConfig).withGoalThreshold(0.01);
final CounterfactualExplainer counterfactualExplainer = new CounterfactualExplainer(counterfactualConfig);
PredictionInput input = new PredictionInput(features);
PredictionOutput output = new PredictionOutput(goal);
Prediction prediction = new CounterfactualPrediction(input, output, null, UUID.randomUUID(), 60L);
final CounterfactualResult counterfactualResult = counterfactualExplainer.explainAsync(prediction, model).get(Config.INSTANCE.getAsyncTimeout(), Config.INSTANCE.getAsyncTimeUnit());
List<Output> cfOutputs = counterfactualResult.getOutput().get(0).getOutputs();
assertTrue(counterfactualResult.isValid());
assertEquals("inputsAreValid", cfOutputs.get(0).getName());
assertTrue((Boolean) cfOutputs.get(0).getValue().getUnderlyingObject());
assertEquals("canRequestLoan", cfOutputs.get(1).getName());
assertTrue((Boolean) cfOutputs.get(1).getValue().getUnderlyingObject());
assertEquals("my-scoring-function", cfOutputs.get(2).getName());
assertEquals(0.6, ((BigDecimal) cfOutputs.get(2).getValue().getUnderlyingObject()).doubleValue(), 0.05);
List<CounterfactualEntity> entities = counterfactualResult.getEntities();
assertEquals("age", entities.get(0).asFeature().getName());
assertEquals(40, entities.get(0).asFeature().getValue().asNumber());
assertEquals("hasReferral", entities.get(1).asFeature().getName());
assertTrue((Boolean) entities.get(1).asFeature().getValue().getUnderlyingObject());
assertEquals("monthlySalary", entities.get(2).asFeature().getName());
assertTrue(entities.get(2).asFeature().getValue().asNumber() > 6000);
}
use of org.optaplanner.core.config.solver.SolverConfig in project optaplanner-quickstarts by kiegroup.
the class TimeTableApp method main.
public static void main(String[] args) {
SolverFactory<TimeTable> solverFactory = SolverFactory.create(new SolverConfig().withSolutionClass(TimeTable.class).withEntityClasses(Lesson.class).withConstraintProviderClass(TimeTableConstraintProvider.class).withTerminationSpentLimit(Duration.ofSeconds(5)));
// Load the problem
TimeTable problem = generateDemoData();
// Solve the problem
Solver<TimeTable> solver = solverFactory.buildSolver();
TimeTable solution = solver.solve(problem);
// Visualize the solution
printTimetable(solution);
}
use of org.optaplanner.core.config.solver.SolverConfig in project kie-benchmarks by kiegroup.
the class NurseRosteringBenchmark method createSolver.
@Override
public Solver<NurseRoster> createSolver() {
SolverConfig solverConfig = Examples.NURSE_ROSTERING.getSolverConfigFromXml();
SolverFactory<NurseRoster> solverFactory = SolverFactory.create(solverConfig);
return solverFactory.buildSolver();
}
use of org.optaplanner.core.config.solver.SolverConfig in project kie-benchmarks by kiegroup.
the class AbstractMultithreadedSolvingScalabilityBenchmark method createSolver.
@Override
public Solver<Solution_> createSolver() {
SolverConfig solverConfig = getBaseSolverConfig();
solverConfig.setMoveThreadCount(moveThreadCount);
LocalSearchPhaseConfig localSearchPhaseConfig = new LocalSearchPhaseConfig();
localSearchPhaseConfig.setMoveSelectorConfig(getMoveSelectorConfig());
localSearchPhaseConfig.setAcceptorConfig(getAcceptorConfig());
localSearchPhaseConfig.setForagerConfig(new LocalSearchForagerConfig());
localSearchPhaseConfig.getForagerConfig().setAcceptedCountLimit(getAcceptedCountLimit());
localSearchPhaseConfig.setTerminationConfig(getTerminationConfig());
solverConfig.setPhaseConfigList(Collections.singletonList(localSearchPhaseConfig));
SolverFactory<Solution_> solverFactory = SolverFactory.create(solverConfig);
return solverFactory.buildSolver();
}
Aggregations