Java library and command-line application for converting Scikit-Learn pipelines to PMML.
- Supported Estimator and Transformer types:
- Clustering:
- Composite Estimators:
- Matrix Decomposition:
- Discriminant Analysis:
- Dummies:
- Ensemble Methods:
ensemble.AdaBoostRegressorensemble.BaggingClassifierensemble.BaggingRegressorensemble.ExtraTreesClassifierensemble.ExtraTreesRegressorensemble.GradientBoostingClassifierensemble.GradientBoostingRegressorensemble.HistGradientBoostingClassifierensemble.HistGradientBoostingRegressorensemble.IsolationForestensemble.RandomForestClassifierensemble.RandomForestRegressorensemble.StackingClassifierensemble.StackingRegressorensemble.VotingClassifierensemble.VotingRegressor
- Feature Extraction:
- Feature Selection:
feature_selection.GenericUnivariateSelect(only viasklearn2pmml.SelectorProxy)feature_selection.RFE(only viasklearn2pmml.SelectorProxy)feature_selection.RFECV(only viasklearn2pmml.SelectorProxy)feature_selection.SelectFdr(only viasklearn2pmml.SelectorProxy)feature_selection.SelectFpr(only viasklearn2pmml.SelectorProxy)feature_selection.SelectFromModel(either directly or viasklearn2pmml.SelectorProxy)feature_selection.SelectFwe(only viasklearn2pmml.SelectorProxy)feature_selection.SelectKBest(either directly or viasklearn2pmml.SelectorProxy)feature_selection.SelectPercentile(only viasklearn2pmml.SelectorProxy)feature_selection.VarianceThreshold(only viasklearn2pmml.SelectorProxy)
- Impute:
- Isotonic regression:
- Generalized Linear Models:
linear_model.ARDRegressionlinear_model.BayesianRidgelinear_model.ElasticNetlinear_model.ElasticNetCVlinear_model.GammaRegressorlinear_model.HuberRegressorlinear_model.Larslinear_model.LarsCVlinear_model.Lassolinear_model.LassoCVlinear_model.LassoLarslinear_model.LassoLarsCVlinear_model.LinearRegressionlinear_model.LogisticRegressionlinear_model.LogisticRegressionCVlinear_model.OrthogonalMatchingPursuitlinear_model.OrthogonalMatchingPursuitCVlinear_model.PoissonRegressorlinear_model.Ridgelinear_model.RidgeCVlinear_model.RidgeClassifierlinear_model.RidgeClassifierCVlinear_model.SGDClassifierlinear_model.SGDRegressorlinear_model.TheilSenRegressor
- Model Selection:
- Multiclass classification:
- Naive Bayes:
- Nearest Neighbors:
- Pipelines:
- Neural network models:
- Preprocessing and Normalization:
preprocessing.Binarizerpreprocessing.FunctionTransformerpreprocessing.Imputerpreprocessing.LabelBinarizerpreprocessing.LabelEncoderpreprocessing.MaxAbsScalerpreprocessing.MinMaxScalerpreprocessing.OneHotEncoderpreprocessing.OrdinalEncoderpreprocessing.PolynomialFeaturespreprocessing.RobustScalerpreprocessing.StandardScaler
- Support Vector Machines:
- Decision Trees:
- Supported third-party Estimator and Transformer types:
- Category Encoders:
- H2O.ai:
- Imbalanced-Learn (imblearn):
imblearn.combine.SMOTEENNimblearn.combine.SMOTETomekimblearn.ensemble.BalancedBaggingClassifierimblearn,ensemble,BalancedRandomForestClassifierimblearn.over_sampling.ADASYNimblearn.over_sampling.BorderlineSMOTEimblearn.over_sampling.KMeansSMOTEimblearn.over_sampling.RandomOverSamplerimblearn.over_sampling.SMOTEimblearn.over_sampling.SMOTENCimblearn.over_sampling.SVMSMOTEimblearn.pipeline.Pipelineimblearn.under_sampling.AllKNNimblearn.under_sampling.ClusterCentroidsimblearn.under_sampling.CondensedNearestNeighbourimblearn.under_sampling.EditedNearestNeighboursimblearn.under_sampling.InstanceHardnessThresholdimblearn.under_sampling.NearMissimblearn.under_sampling.NeighbourhoodCleaningRuleimblearn.under_sampling.OneSidedSelectionimblearn.under_sampling.RandomUnderSamplerimblearn.under_sampling.RepeatedEditedNearestNeighboursimblearn.under_sampling.TomekLinks
- LightGBM:
- SkLearn2PMML:
sklearn2pmml.EstimatorProxysklearn2pmml.SelectorProxysklearn2pmml.decoration.Aliassklearn2pmml.decoration.CategoricalDomainsklearn2pmml.decoration.ContinuousDomainsklearn2pmml.decoration.DateDomainsklearn2pmml.decoration.DateTimeDomainsklearn2pmml.decoration.MultiDomainsklearn2pmml.decoration.OrdinalDomainsklearn2pmml.ensemble.GBDTLMRegressor- The GBDT side: All Scikit-Learn decision tree ensemble regressors,
LGBMRegressor,XGBRegressor,XGBRFRegressor. - The LM side: A Scikit-Learn linear regressor (eg.
ElasticNet,LinearRegression,SGDRegressor).
- The GBDT side: All Scikit-Learn decision tree ensemble regressors,
sklearn2pmml.ensemble.GBDTLRClassifier- The GBDT side: All Scikit-Learn decision tree ensemble classifiers,
LGBMClassifier,XGBClassifier,XGBRFClassifier. - The LR side: A Scikit-Learn binary linear classifier (eg.
LinearSVC,LogisticRegression,SGDClassifier).
- The GBDT side: All Scikit-Learn decision tree ensemble classifiers,
sklearn2pmml.ensemble.SelectFirstClassifiersklearn2pmml.ensemble.SelectFirstRegressorsklearn2pmml.feature_selection.SelectUniquesklearn2pmml.pipeline.PMMLPipelinesklearn2pmml.preprocessing.Aggregatorsklearn2pmml.preprocessing.CastTransformersklearn2pmml.preprocessing.ConcatTransformersklearn2pmml.preprocessing.CutTransformersklearn2pmml.preprocessing.DaysSinceYearTransformersklearn2pmml.preprocessing.ExpressionTransformer- Ternary conditional expression
<expression_true> if <condition> else <expression_false>. - Array indexing expressions
X[<column index>]andX[<column name>]. - String concatenation expressions.
- String slicing expressions
<str>[<start>:<stop>]. - Arithmetic operators
+,-,*,/and%. - Identity comparison operators
is Noneandis not None. - Comparison operators
in <list>,not in <list>,<=,<,==,!=,>and>=. - Logical operators
and,orandnot. - Value missingness check functions
pandas.isnullandpandas.notnull. - Numpy universal functions.
- String functions
lower,upperandstrip. - String length function
len(<str>)
- Ternary conditional expression
sklearn2pmml.preprocessing.IdentityTransformersklearn2pmml.preprocessing.LookupTransformersklearn2pmml.preprocessing.MatchesTransformersklearn2pmml.preprocessing.MultiLookupTransformersklearn2pmml.preprocessing.PMMLLabelBinarizersklearn2pmml.preprocessing.PMMLLabelEncodersklearn2pmml.preprocessing.PowerFunctionTransformersklearn2pmml.preprocessing.ReplaceTransformersklearn2pmml.preprocessing.SecondsSinceMidnightTransformersklearn2pmml.preprocessing.SecondsSinceYearTransformersklearn2pmml.preprocessing.StringNormalizersklearn2pmml.preprocessing.SubstringTransformersklearn2pmml.preprocessing.WordCountTransformersklearn2pmml.preprocessing.h2o.H2OFrameCreatorsklearn2pmml.preprocessing.scipy.BSplineTransformersklearn2pmml.ruleset.RuleSetClassifier
- Sklearn-Pandas:
sklearn_pandas.CategoricalImputersklearn_pandas.DataFrameMapper
- TPOT:
tpot.builtins.stacking_estimator.StackingEstimator
- XGBoost:
- Production quality:
- Complete test coverage.
- Fully compliant with the JPMML-Evaluator library.
- Python 2.7, 3.4 or newer.
scikit-learn0.16.0 or newer.sklearn-pandas0.0.10 or newer.sklearn2pmml0.14.0 or newer.
Validating Python installation:
import sklearn, sklearn.externals.joblib, sklearn_pandas, sklearn2pmml
print(sklearn.__version__)
print(sklearn.externals.joblib.__version__)
print(sklearn_pandas.__version__)
print(sklearn2pmml.__version__)- Java 1.8 or newer.
Enter the project root directory and build using Apache Maven:
mvn clean install
The build produces an executable uber-JAR file target/jpmml-sklearn-executable-1.6-SNAPSHOT.jar.
A typical workflow can be summarized as follows:
- Use Python to train a model.
- Serialize the model in
pickledata format to a file in a local filesystem. - Use the JPMML-SkLearn command-line converter application to turn the pickle file to a PMML file.
Loading data to a pandas.DataFrame object:
import pandas
df = pandas.read_csv("Iris.csv")
iris_X = df[df.columns.difference(["Species"])]
iris_y = df["Species"]First, creating a sklearn_pandas.DataFrameMapper object, which performs column-oriented feature engineering and selection work:
from sklearn_pandas import DataFrameMapper
from sklearn.preprocessing import StandardScaler
from sklearn2pmml.decoration import ContinuousDomain
column_preprocessor = DataFrameMapper([
(["Sepal.Length", "Sepal.Width", "Petal.Length", "Petal.Width"], [ContinuousDomain(), StandardScaler()])
])Second, creating Transformer and Selector objects, which perform table-oriented feature engineering and selection work:
from sklearn.decomposition import PCA
from sklearn.feature_selection import SelectKBest
from sklearn.pipeline import Pipeline
from sklearn2pmml import SelectorProxy
table_preprocessor = Pipeline([
("pca", PCA(n_components = 3)),
("selector", SelectorProxy(SelectKBest(k = 2)))
])Please note that stateless Scikit-Learn selector objects need to be wrapped into an sklearn2pmml.SelectprProxy object.
Third, creating an Estimator object:
from sklearn.tree import DecisionTreeClassifier
classifier = DecisionTreeClassifier(min_samples_leaf = 5)Combining the above objects into a sklearn2pmml.pipeline.PMMLPipeline object, and running the experiment:
from sklearn2pmml.pipeline import PMMLPipeline
pipeline = PMMLPipeline([
("columns", column_preprocessor),
("table", table_preprocessor),
("classifier", classifier)
])
pipeline.fit(iris_X, iris_y)Embedding model verification data:
pipeline.verify(iris_X.sample(n = 15))Storing the fitted PMMLPipeline object in pickle data format:
from sklearn.externals import joblib
joblib.dump(pipeline, "pipeline.pkl.z", compress = 9)Please see the test script file main.py for more classification (binary and multi-class) and regression workflows.
Converting the pipeline pickle file pipeline.pkl.z to a PMML file pipeline.pmml:
java -jar target/jpmml-sklearn-executable-1.6-SNAPSHOT.jar --pkl-input pipeline.pkl.z --pmml-output pipeline.pmml
Getting help:
java -jar target/jpmml-sklearn-executable-1.6-SNAPSHOT.jar --help
Up-to-date:
- Converting Scikit-Learn based Imbalanced-Learn (imblearn) pipelines to PMML documents
- Extending Scikit-Learn with date and datetime features
- Extending Scikit-Learn with feature specifications
- Converting logistic regression models to PMML documents
- Stacking Scikit-Learn, LightGBM and XGBoost models
- Converting Scikit-Learn hyperparameter-tuned pipelines to PMML documents
- Extending Scikit-Learn with GBDT plus LR ensemble (GBDT+LR) model type
- Converting Scikit-Learn based TPOT automated machine learning (AutoML) pipelines to PMML documents
- Converting Scikit-Learn based LightGBM pipelines to PMML documents
- Extending Scikit-Learn with business rules (BR) model type
Slightly outdated:
JPMML-SkLearn is licensed under the terms and conditions of the GNU Affero General Public License, Version 3.0.
If you would like to use JPMML-SkLearn in a proprietary software project, then it is possible to enter into a licensing agreement which makes JPMML-SkLearn available under the terms and conditions of the BSD 3-Clause License instead.
JPMML-SkLearn is developed and maintained by Openscoring Ltd, Estonia.
Interested in using Java PMML API software in your company? Please contact [email protected]