首页 > 解决方案 > 无法在 Azure Synapse 笔记本中将 ML 模型转储为 pickle 文件

问题描述

我在 Azure Synapse 中使用 pyspark 制作了推荐引擎 ALS 模型。我正在尝试将模型转储到 pickle 文件中并将其注册到 Azure ML Studio。

这就是我尝试制作泡菜文件的方式:

 from sklearn.externals import joblib
 joblib.dump(model, 'recommendation_engine.pkl')

但是我收到此错误:

Py4JError: An error occurred while calling o220.__getstate__. Trace:
py4j.Py4JException: Method __getstate__([]) does not exist
    at py4j.reflection.ReflectionEngine.getMethod(ReflectionEngine.java:318)
    at py4j.reflection.ReflectionEngine.getMethod(ReflectionEngine.java:326)
    at py4j.Gateway.invoke(Gateway.java:274)
    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)

这些是模型参数:

als = ALS(
    maxIter=MAX_ITER, 
    rank=RANK,
    regParam=REG_PARAM, 
    userCol=COL_USER, 
    itemCol=COL_ITEM, 
    ratingCol=COL_RATING, 
    coldStartStrategy="drop"
)

model = als.fit(dfs_train)

有什么建议么?

标签: azuremachine-learningpysparkpickleazure-synapse

解决方案


推荐阅读