MLeap нарушен с помощью Skicit-learn при сериализации: объект не имеет атрибута input_features

Question

У меня проблема с MLeap 0.16 и Python 3, когда я пытаюсь сериализовать модель. Вот мой код:

from mleap.sklearn.logistic import LogisticRegression
from sklearn.datasets import load_iris

X, y = load_iris(return_X_y=True)
clf = LogisticRegression(random_state=0).fit(X, y)

clf.serialize_to_bundle("path", "irismodel")

ошибка:

AttributeError: 'LogisticRegression' object has no attribute 'input_features'

Кто-нибудь нашел обходной путь?

Answer 1

Я нашел решение.

clf.mlinit(input_features="features", prediction_column="prediction") 

отсутствует.

Для этого также можно использовать конвейер:

from mleap.sklearn.logistic import LogisticRegression
from sklearn.datasets import load_iris
from mleap.sklearn.pipeline import Pipeline

X, y = load_iris(return_X_y=True)
logistic = LogisticRegression(random_state=0)
logistic.mlinit(input_features="features", prediction_column="prediction")
pipeline = Pipeline([("log", logistic)])
clf = pipeline.fit(X, y)

clf.mlinit()

clf.serialize_to_bundle("/dbfs/endpath", "model.json")