Переименовать функции в оценщике TensorFlow - PullRequest
0 голосов
/ 16 января 2019

Я использую постоянный оценщик и читаю данные из источника, который я не могу контролировать. Источник хранит объекты с помощью snake_case, тогда как то, что я даю, функция предикта всегда входит в camelCase и также находится вне моего контроля.

Грубо говоря, я обучаю модель и экспортирую ее, чтобы потом прочитать ее:

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

Есть ли способ переименовать функции (от случая змеи до верблюда), когда я обучаю модель (как псевдоним или что-то в этом роде) или впоследствии, когда я ее сохраняю?

1 Ответ

0 голосов
/ 16 января 2019

Вы можете использовать что-то вроде этого

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

Они переименовывают переменные, хранящиеся в контрольной точке, после итерации переменных, возвращаемых tf.contrib.framework.list_variables()

...