首页 > 解决方案 > 如何使用 Tensorflow SavedModel 保存 RandomizedSearchCV 对象的最佳估计器?

问题描述

我正在尝试使用 SavedModel 格式从 RandomizedSearchCV 对象中保存最佳估计器。我知道我们也可以使用保存它,joblib.dump但我需要它是 SavedModel 格式。

通常,我总是使用:

...
..
.
model.fit(X,y,epochs=1000)
model.save_model("MyModel")

保存我的训练模型(没有超参数优化)

但是,这一次当我尝试时它不起作用:

.....
...
..
result = RandomizedSearchCV(*args)
result.best_estimator_.save_model("MyModel")

错误表明:

RandomizedSearchCV object does not have attribute saved_model

任何帮助将不胜感激!

提前致谢。

标签: pythontensorflowkerasscikit-learnpython-3.7

解决方案


推荐阅读