преобразование клиента в PMML с помощью SkLearn2PMML-Plugin на стороне Java - PullRequest
0 голосов
/ 27 ноября 2018

Мне известен проект SkLearn2PMML-Plugin в github (https://github.com/jpmml/sklearn2pmml-plugin/blob/master/README.md).. Но у меня мало опыта в Java. Может ли кто-нибудь помочь мне написать плагин Java для моего преобразователя функций. Ниже приведен мой преобразователь функций.

class FeatureSelector(TransformerMixin):
'''A transformer for extracting certain column(s)'''
def __init__(self, cols):
    self.cols = cols

def fit(self, X, y=None, **fit_params):
    return self

def transform(self, X, **transform_params):
    return X[self.cols]




class ModelTransformer(TransformerMixin):

def __init__(self, model):
    self.model = model

def fit(self, *args, **kwargs):
    self.model.fit(*args, **kwargs)
    return self

def transform(self, X, **transform_params):
    return pd.DataFrame(self.model.predict(X))

1 Ответ

0 голосов
/ 27 ноября 2018

Функциональность FeatureSelector можно получить с помощью преобразования sklearn2pmml.preprocessing.ExpressionTransformer:

selector = ExpressionTransformer("X[0]")

Функциональность ModelTransformer немного сложнее, но, безусловно, выполнима.В следующий раз, пожалуйста, подумайте об открытии запроса на добавление в проекте SkLearn2PMML напрямую (вместо того, чтобы просить SO написать для вас код): https://github.com/jpmml/sklearn2pmml/issues/118

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...