gpt4 book ai didi

python - 重命名 TensorFlow 估算器中的特征

转载 作者:太空宇宙 更新时间:2023-11-03 21:21:22 24 4
gpt4 key购买 nike

我正在使用预装估算器并从我无法控制的源读取数据。源代码使用snake_case存储特征,而我给出的预测函数总是以camelCase形式出现,并且也超出了我的控制范围。

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

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

有没有办法在我训练模型时(作为别名或类似的东西)或之后保存模型时重命名特征(从蛇形命名为驼峰命名)?

最佳答案

你可以使用这样的东西

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

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

关于python - 重命名 TensorFlow 估算器中的特征,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54215329/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com