use of org.jpmml.evaluator.ResultField in project jpmml-sparkml by jpmml.
the class AssociationRulesTest method evaluateFPGrowthShopping.
@Test
public void evaluateFPGrowthShopping() throws Exception {
Predicate<ResultField> predicate = (resultField -> true);
Equivalence<Object> equivalence = getEquivalence();
try (SparkMLTestBatch batch = (SparkMLTestBatch) createBatch(FP_GROWTH, SHOPPING, predicate, equivalence)) {
PMML pmml = batch.getPMML();
Model model = Iterables.getOnlyElement(pmml.getModels());
assertTrue(model instanceof AssociationModel);
}
}
Aggregations