diff options
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/xgboost/XGBoostImportEvaluationTestCase.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/xgboost/XGBoostImportEvaluationTestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/xgboost/XGBoostImportEvaluationTestCase.java index 78137352512..ec2498b3923 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/xgboost/XGBoostImportEvaluationTestCase.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/xgboost/XGBoostImportEvaluationTestCase.java @@ -43,10 +43,10 @@ public class XGBoostImportEvaluationTestCase { optimizer.optimize(expression, (ContextIndex)context); assertTrue(expression.getRoot() instanceof GBDTForestNode); -// assertXGBoostEvaluation(1.0, expression, features(context, "f1", 0.0, "f2", 0.0)); -// assertXGBoostEvaluation(2.0, expression, features(context, "f1", 0.0, "f2", 1.0)); -// assertXGBoostEvaluation(3.0, expression, features(context, "f1", 1.0, "f2", 0.0)); -// assertXGBoostEvaluation(4.0, expression, features(context, "f1", 1.0, "f2", 1.0)); + assertXGBoostEvaluation(1.0, expression, features(context, "f1", 0.0, "f2", 0.0)); + assertXGBoostEvaluation(2.0, expression, features(context, "f1", 0.0, "f2", 1.0)); + assertXGBoostEvaluation(3.0, expression, features(context, "f1", 1.0, "f2", 0.0)); + assertXGBoostEvaluation(4.0, expression, features(context, "f1", 1.0, "f2", 1.0)); assertXGBoostEvaluation(5.0, expression, features(context, "f1", 0.0)); assertXGBoostEvaluation(6.0, expression, features(context, "f1", 1.0)); assertXGBoostEvaluation(7.0, expression, features(context, "f2", 0.0)); |