PicklingError:无法pickle<function>,Tensorflow

2024-09-27 22:31:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试为我的模型创建一个.joblib,以便在另一个项目中使用,但是,每当我尝试这样做时,总是会出现此错误

错误:

PicklingError: Can't pickle <function DNNClassifierV2.__init__.<locals>._model_fn at 0x000001D45EB1C160>: it's not found as tensorflow_estimator.python.estimator.canned.dnn.DNNClassifierV2.__init__.<locals>._model_fn

我在这行代码中得到错误:

dump(classifier, 'Pokemon.joblib')

其中,分类器为:

classifier = tf.estimator.DNNClassifier(
    feature_columns = my_feature_columns,
    hidden_units=[30, 10],
    n_classes = 18
)

我是新手,所以我不确定这是否是最好的方法


Tags: columns项目模型modelinit错误canfeature

热门问题