NoSuchMethodException: org.apache.spark.ml.classification.GBTClassificationModel в загрузке модели Pyspark

Я обучил модель в pyspark

##Model
gbt = GBTClassifier(maxIter=10)
gbtModel = gbt.fit(train)
predictions = gbtModel.transform(test)

Здесь я сохраняю пайплайн и модель

#Save pipeline
pipelineModel.write().overwrite().save("s3://data-production/pipelineModel_v1")

#Save Model
gbtModel.save("s3://data-production/first_trade.model_v0")

Сейчас в производстве / будущих наборах данных, я загружаю конвейер и модель

pipelineModel = PipelineModel.load("s3://data-production/pipelineModel_v1")

new_test= pipelineModel.transform(new_df1)

model = GBTClassifier.load("s3://data-production/first_trade.model_v0")

Я получаю эту ошибку после загрузки модели

Py4JJavaError: An error occurred while calling o4701.load.
: java.lang.NoSuchMethodException: org.apache.spark.ml.classification.GBTClassificationModel.<init>(java.lang.String)
    at java.lang.Class.getConstructor0(Class.java:3082)
    at java.lang.Class.getConstructor(Class.java:1825)
    at org.apache.spark.ml.util.DefaultParamsReader.load(ReadWrite.scala:496)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
    at java.lang.reflect.Method.invoke(Method.java:498)
    at py4j.reflection.MethodInvoker.invoke(MethodInvoker.java:244)
    at py4j.reflection.ReflectionEngine.invoke(ReflectionEngine.java:357)
    at py4j.Gateway.invoke(Gateway.java:282)
    at py4j.commands.AbstractCommand.invokeMethod(AbstractCommand.java:132)
    at py4j.commands.CallCommand.execute(CallCommand.java:79)
    at py4j.GatewayConnection.run(GatewayConnection.java:238)
    at java.lang.Thread.run(Thread.java:748)

(<class 'py4j.protocol.Py4JJavaError'>, Py4JJavaError('An error occurred while calling o4701.load.\n', JavaObject id=o4702), <traceback object at 0x7f247a3eb9c8>)

pyspark apache-spark-ml
person Shubh    schedule 14.02.2021    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Сохраненная модель, по сути, представляет собой сериализованную версию вашего обученного GBTClassifier. Для десериализации модели вам также понадобятся исходные классы в производственном коде. Добавьте эту строку в набор операторов импорта.

from pyspark.ml.classification import GBTClassifier, GBTClassificationModel
person nimish    schedule 18.02.2021