Updated test resources

jpmml · Jun 1, 2024 · 3c6f90a · 3c6f90a
1 parent cc0bb4b
commit 3c6f90a
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 0 deletions.
diff --git a/pmml-biogeme/src/main/resources/META-INF/biogeme2pmml.properties b/pmml-biogeme/src/main/resources/META-INF/biogeme2pmml.properties
@@ -6,6 +6,7 @@ biogeme.expressions.comparison_expressions.Equal = biogeme.expressions.Equal
 biogeme.expressions.comparison_expressions.NotEqual = biogeme.expressions.NotEqual
 biogeme.expressions.elementary_expressions.Beta = biogeme.expressions.Beta
 biogeme.expressions.elementary_expressions.Variable = biogeme.expressions.Variable
+biogeme.expressions.logit_expressions._bioLogLogit = biogeme.models.LogLogit
 biogeme.expressions.logit_expressions._bioLogLogitFullChoiceSet = biogeme.models.LogLogit
 biogeme.expressions.numeric_expressions.Numeric = biogeme.expressions.Numeric
 biogeme.results.bioResults = biogeme.results.Results

diff --git a/pmml-biogeme/src/test/resources/main.py b/pmml-biogeme/src/test/resources/main.py
@@ -62,6 +62,8 @@ def predict(proba_func, experiment):
 
 availability[1] = Variable("AV_CAR")
 
+experiment["model"] = loglogit(V, availability, Choice)
+
 store_pkl(experiment, "MNLAvOptima")
 
 prediction = predict(lambda x: logit(V, availability, x), experiment)

diff --git a/pmml-biogeme/src/test/resources/pkl/MNLAvOptima.pkl b/pmml-biogeme/src/test/resources/pkl/MNLAvOptima.pkl
diff --git a/pmml-biogeme/src/test/resources/pkl/MNLOptima.pkl b/pmml-biogeme/src/test/resources/pkl/MNLOptima.pkl