美文网首页
pyspark自定义的pipeline无法保存

pyspark自定义的pipeline无法保存

作者: 一只当归 | 来源:发表于2020-12-21 20:26 被阅读0次

自定义pipeline方法feature_engineering进行保存时出现错误
ValueError: ('Pipeline write will fail on this pipeline because stage %s of type %s is not MLWritable', 'feature_engineering_94caf721906c', <class 'main.feature_engineering'>)

解决方法:
类继承DefaultParamsReadable, DefaultParamsWritable类,会自动载入保存以及读取方法

from pyspark.ml.util import DefaultParamsReadable, DefaultParamsWritable
class feature_engineering(Transformer,DefaultParamsReadable, DefaultParamsWritable):

相关文章

网友评论

      本文标题:pyspark自定义的pipeline无法保存

      本文链接:https://www.haomeiwen.com/subject/rjnrnktx.html