Exporting PMML for Class BernouilliNB

It is easy and simple to build a naive Bayes model using scikit-learn. If you haven't familiarized with it yet, we recommend that you take a look at the following scikit-learn documentation:

Class BernouilliNB

Once the naive Bayes model is built, all you need to do is to export the model parameters to a .txt file which can then be given to Py2PMML so that it generates the equivalent PMML code for your model.

The .txt file needs to follow a strict sequence. This is defined as follows:

  1. Model type: The name of the scikit-learn class used to build the model (string). In this case, "GaussianNB" (without quotation marks)
  2. Model name: This is the name you are given your model in PMML (string)
  3. Function type: In this case, "classification" (no quotation marks - string)
  4. Number of input fields (integer)
  5. Input fields (one entry per line): name, data type, operational type, missing value replacement, missing value treatment, and invalid value treatment (sequence of strings, comma separated). Make sure each string follows the PMML nomenclature as outlined below.
  6. Number of target categories (integer)
  7. The name of each of the target categories (sequence of strings, one name per line)
  8. Class log prior (log probability of each class - float, one value per target class/line)
  9. Feature log prob (float, one value per input feature/line)

The following Python code exemplifies the writing of the .txt file for a simple naive Bayes model built using random data (5 classes, 20 input features).


For this code, the resulting .txt file is written to file "BernouilliNBModel.txt" which will contain the following information:


PMML Nomenclature and Input Field Information

For each input field, the .txt file is required to have the following important information which must follow the PMML nomenclature:

  1. Data type: "double", "float", "integer", or "string". This information is required in PMML.
  2. Operational type: "continuous" or "categorical". This information is required in PMML.
  3. Missing value replacement: If the field is continuous, this value should be a number. For example, you may want to replace any possible missing values for input field "age" by value "31", the mean value for age in your historical data. Use value "NA" if you do not want to this information to be part of the PMML code.
  4. Missing value treatment: PMML uses this attribute for information only. Possible values are: "asMean", "asMode", "asMedian", "asValue", and "asIs". Use value "NA" if you do not want this information to be part of the PMML code.
  5. Invalid value treatment: Used to deal with invalid values. For example, if an invalid value is encountered, you may want to treat it the same way you are treating missing values. Possible values are: "returnInvalid" (default), "asIs", "asMissing". Use value "NA" if you do not want this information to be part of the PMML code.


Go ahead and give it a try. Pass the .txt file through Py2PMML to obtain the equivalent PMML code for the model. With the PMML file in hand, your model can be executed anywhere using one of the Zementis scoring products.


Article is closed for comments.
Powered by Zendesk