combust/mleap

Logistic Regression prediction probabilities

Open

#464 建立於 2018年12月22日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)Scala (312 fork)batch import
help wantedwaiting for input

倉庫指標

Star
 (1,461 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

Currently sklearn logistic classifier only supports predictions.

https://github.com/neilsummers/mleap/blob/master/python/mleap/sklearn/logistic.py

    def serialize_to_bundle(self, transformer, path, model_name):

        # compile tuples of model attributes to serialize
        attributes = list()
        attributes.append(('intercept', transformer.intercept_.tolist()[0]))
        attributes.append(('coefficients', transformer.coef_.tolist()[0]))
        attributes.append(('num_classes', 2)) # TODO: get number of classes from the transformer

        # define node inputs and outputs
        inputs = [{
                  "name": transformer.input_features,
                  "port": "features"
                }]

        outputs = [{
                  "name": transformer.prediction_column,
                  "port": "prediction"
                }]

        self.serialize(transformer, path, model_name, attributes, inputs, outputs)

Want to be able to access the probabilities from the classifier too. This is already supported in the MLeap logistic model. Just need to add the appropriate outputs. I will create a PR with the change we made to access this for our own production model.

貢獻者指南