Updated test resources

jpmml · Jan 25, 2024 · 0a5c6f6 · 0a5c6f6
1 parent ed5e025
commit 0a5c6f6
Show file tree

Hide file tree

Showing 28 changed files with 10,746 additions and 9,945 deletions.
diff --git a/pmml-sklearn-xgboost/src/test/java/org/jpmml/sklearn/xgboost/testing/CategoricalTest.java b/pmml-sklearn-xgboost/src/test/java/org/jpmml/sklearn/xgboost/testing/CategoricalTest.java
@@ -68,17 +68,22 @@ public List<Map<String, Object>> getOptionsMatrix(){
 
 	@Test
 	public void evaluateXGBAuditCat() throws Exception {
-		evaluate("XGB", AUDIT + "Cat", excludeFields(AUDIT_PROBABILITY_FALSE), new FloatEquivalence(12)); // XXX
+		evaluate("XGB", AUDIT + "Cat", excludeFields(AUDIT_PROBABILITY_FALSE), new FloatEquivalence(48 + 8)); // XXX
 	}
 
 	@Test
 	public void evaluateXGBAuditCatNA() throws Exception {
-		evaluate("XGB", AUDIT + "CatNA", excludeFields(AUDIT_PROBABILITY_TRUE), new FloatEquivalence(12));
+		evaluate("XGB", AUDIT + "CatNA", excludeFields(AUDIT_PROBABILITY_TRUE), new FloatEquivalence(56 + 8)); // XXX
 	}
 
 	@Test
 	public void evaluateXGBAutoCat() throws Exception {
-		evaluate("XGB", AUTO + "Cat", new FloatEquivalence(8));
+		evaluate("XGB", AUTO + "Cat", new FloatEquivalence(8 + 4));
+	}
+
+	@Test
+	public void evaluateMultiXGBAutoCat() throws Exception {
+		evaluate("MultiXGB", AUTO + "Cat", new FloatEquivalence(12));
 	}
 
 	@Test
@@ -88,17 +93,22 @@ public void evaluateXGBAutoCatNA() throws Exception {
 
 	@Test
 	public void evaluateXGBRFAuditCat() throws Exception {
-		evaluate("XGBRF", AUDIT + "Cat", excludeFields(AUDIT_PROBABILITY_FALSE), new FloatEquivalence(8));
+		evaluate("XGBRF", AUDIT + "Cat", excludeFields(AUDIT_PROBABILITY_FALSE), new FloatEquivalence(24));
 	}
 
 	@Test
 	public void evaluateXGBRFAuditCatNA() throws Exception {
-		evaluate("XGBRF", AUDIT + "CatNA", excludeFields(AUDIT_PROBABILITY_TRUE), new FloatEquivalence(8));
+		evaluate("XGBRF", AUDIT + "CatNA", excludeFields(AUDIT_PROBABILITY_TRUE), new FloatEquivalence(16 + 4));
 	}
 
 	@Test
 	public void evaluateXGBRFAutoCat() throws Exception {
-		evaluate("XGBRF", AUTO + "Cat", new FloatEquivalence(4));
+		evaluate("XGBRF", AUTO + "Cat", new FloatEquivalence(8 + 2));
+	}
+
+	@Test
+	public void evaluateMultiXGBRFAutoCat() throws Exception {
+		evaluate("MultiXGBRF", AUTO + "Cat", new FloatEquivalence(32 + 8));
 	}
 
 	@Test