Преобразование клиента в PMML с помощью SkLearn2PMML-Plugin на стороне Java

Question

Я знал проект 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))

Answer 1

Вы можете достичь FeatureSelector функциональность с использованием sklearn2pmml.preprocessing.ExpressionTransformer трансформация:

selector = ExpressionTransformer("X[0]")

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