Supported models for import
Learn more about the PMML and H2O models that you can import to Prediction Studio.
Table of contents
This article covers the following topics:
Supported PMML model types
Pega Platform uses a specific implementation of the PMML format, which means that some PMML features and models are not supported in the Predictive Model rule.
You can import models from the following PMML versions:
- 3.0
- 3.1
- 3.2
- 4.0
- 4.1
- 4.2
- 4.2.1
- 4.3
- 4.4
You can import PMML models that use the following algorithms:
- Anomaly detection
- Clustering
- Decision tree
- General regression
- K-nearest neighbors
- Naive Bayes
- Neural network
- Regression
- Ruleset
- Scorecard
- Support Vector Machine
- Ensemble methods (including Random forest and Gradient boosting)
Guidelines and restrictions for importing PMML models
When importing PMML models to Pega Platform, take into account the following guidelines and restrictions:
- Clustering:
- The kind attribute of the
ComparisonMeasureelement can be set to distance or similarity.
- The kind attribute of the
- Decision tree:
- The
functionNameattribute for theTreeModelelement cannot be empty, and has to be set to regression or classification.
- The
- General regression:
- If the
functionNameattribute of theGeneralRegressionModelelement is regression, the model must have exactly one PPMatrix. - The
multinomialLogistic,ordinalMultinomial, andCoxRegressionalgorithms are not supported for the regression mining function. - The
regression,general_linear, andCoxRegressionalgorithms are not supported for the classification mining function.
- If the
- K-nearest neighbors:
- The
opTypeattribute of the input field (DataField) can be set to continuous or categorical. - The kind attribute of the
ComparisonMeasureelement can be set to distance or similarity.
- The
- Naive Bayes models support only one classification mining function.
- Neural network:
- The
miningfunction can be regression or classficiation.
- The
- Regression:
- If the
functionNameattribute of theRegressionModelelement has the value regression, thenormalizationMethodattribute can have one of the following values: none, softmax, logit, or exp. - If the
functionNameattribute of theRegressionModelelement has the value classification, thenormalizationMethodattribute can have one of the following values: none, softmax, logit, loglog, or cloglog.
- If the
- Scorecard:
- The
functionNameattribute is mandatory for theScorecardModelelement.
- The
- Support Vector Machine:
- The
svmRepresentationattribute is mandatory for theSupportVectorMachineModelelement. - The
functionNameattribute for theSupportVectorMachineModelelement cannot be empty and has to be set to regression or classification. - The
probabilityattribute value is not supported for theresultFeatureattribute in theOutputelement.
- The
Unsupported PMML model types
Pega Platform does not support PMML models that use the following algorithms:
- Association rules
- Base line
- Ensemble methods (including Mining and Many-in-one) that contain composite embedded models
- Sequences
- Text
- Time series
Supported H2O model types
You can import H2O-3 and Driverless AI models in .mojo format.
You can import H2O-3 models that use the following algorithms:
- Cox proportional hazards
- Deep learning
- Distributed random Forest
- Generalized linear model
- Gradient boosting machine
- Isolation forest
- K-means clustering
- Naive Bayes classifier
- Stacked ensembles
- Support Vector Machine
- XGBoost