我正在使用固定的估算器,并从无法控制的来源读取数据。源代码使用snake_case存储功能,而我给预测功能提供的功能总是在camelCase中,这也是我无法控制的。
粗略地讲,我训练模型并将其导出,以便稍后可以阅读:
features = ... # snake_case
estimator = tf.estimator(DNNClassifier(feature_columns=features, ...)
...
estimator.export_savedmodel(...)
当我训练模型(作为别名或类似名称)时或在保存模型之后,是否有办法重命名功能(从蛇形到骆驼形)?
答案 0 :(得分:0)
您可以使用类似的东西
https://gist.github.com/batzner/7c24802dd9c5e15870b4b56e22135c96
它们在遍历tf.contrib.framework.list_variables()
返回的变量后重命名存储在检查点中的变量