重命名TensorFlow估算器中的功能

时间:2019-01-16 10:43:55

标签: python tensorflow tensorflow-estimator

我正在使用固定的估算器,并从无法控制的来源读取数据。源代码使用snake_case存储功能,而我给预测功能提供的功能总是在camelCase中,这也是我无法控制的。

粗略地讲,我训练模型并将其导出,以便稍后可以阅读:

features = ... # snake_case
estimator = tf.estimator(DNNClassifier(feature_columns=features, ...)
...
estimator.export_savedmodel(...)

当我训练模型(作为别名或类似名称)时或在保存模型之后,是否有办法重命名功能(从蛇形到骆驼形)?

1 个答案:

答案 0 :(得分:0)

您可以使用类似的东西

https://gist.github.com/batzner/7c24802dd9c5e15870b4b56e22135c96

它们在遍历tf.contrib.framework.list_variables()返回的变量后重命名存储在检查点中的变量