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
ComparisonMeasure
element can be set to distance or similarity.
- The kind attribute of the
- Decision tree:
- The
functionName
attribute for theTreeModel
element cannot be empty, and has to be set to regression or classification.
- The
- General regression:
- If the
functionName
attribute of theGeneralRegressionModel
element is regression, the model must have exactly one PPMatrix. - The
multinomialLogistic
,ordinalMultinomial
, andCoxRegression
algorithms are not supported for the regression mining function. - The
regression
,general_linear
, andCoxRegression
algorithms are not supported for the classification mining function.
- If the
- K-nearest neighbors:
- The
opType
attribute of the input field (DataField
) can be set to continuous or categorical. - The kind attribute of the
ComparisonMeasure
element can be set to distance or similarity.
- The
- Naive Bayes models support only one classification mining function.
- Neural network:
- The
mining
function can be regression or classficiation.
- The
- Regression:
- If the
functionName
attribute of theRegressionModel
element has the value regression, thenormalizationMethod
attribute can have one of the following values: none, softmax, logit, or exp. - If the
functionName
attribute of theRegressionModel
element has the value classification, thenormalizationMethod
attribute can have one of the following values: none, softmax, logit, loglog, or cloglog.
- If the
- Scorecard:
- The
functionName
attribute is mandatory for theScorecardModel
element.
- The
- Support Vector Machine:
- The
svmRepresentation
attribute is mandatory for theSupportVectorMachineModel
element. - The
functionName
attribute for theSupportVectorMachineModel
element cannot be empty and has to be set to regression or classification. - The
probability
attribute value is not supported for theresultFeature
attribute in theOutput
element.
- 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