我正在训练一个DCN模型以达到排名的目的。训练结束后,我使用model.save_weights(filepath)来保存体重。我使用model.load_weights(filepath)在另一个文件中加载权重。在colab上进行培训和保存,在我的本地系统上进行加载保存模型。然而,在colab上所做的预测和在本地系统中所做的预测,即使在编译之后也是不同的,因为相同的输入是不同的。两个版本的tensorflow, tensorflow_recommnders and tensorflow_ranking都是相同的。为了避免这个问题,我尝试用model.save('model.h5
我正在使用contrib.learn.estimator在tensorflow0.12环境中进行预测。
#1. Use a regression Estimator, set n_classes to 0
model = skflow.SKCompat(skflow.Estimator(model_fn=lstm_model, model_dir=LOG_DIR))
#2. create a lstm instance and validation monitor
validation_monitor = skflow.monitors.ValidationMonitor(X, y,