Оценщик Tensorflow: Как я могу получить имя оценки внутри model_fn оценщика? - PullRequest
0 голосов
/ 14 апреля 2020

В оценщиках мы можем передать имя для оценки, либо в estimator.evaluate(..., name=(...)), либо как часть EvalSpec(..., name=(...)). Estimator создает средство составления сводок по умолчанию и записывает сводки в каталог eval_{name}. Эти резюме включают global_steps_per_second и loss.

Я использую два разных набора данных и выполняю оценку дважды каждый с другим именем, а также записываю некоторые тензоры как во время обучения, так и во время оценки. Чтобы записать эти тензоры (или любой файл) в этот каталог, мне нужно имя каталога в пределах model_fn(), поэтому я могу использовать его, чтобы получить действительный каталог eval и правильно зарегистрировать, если mode равен tf.estimator.ModeKeys.PREDICT.

Оценщики имеют .eval_dir метод , однако, как вы можете видеть в source , он просто возвращает eval_{name} при передаче параметра name.

Как я могу получить name оценки внутри model_fn оценки?

...