Мне известен проект 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))