TensorFlow Estimator BestExporter event_file_pattern ничего не делает - PullRequest
Новая
       28

TensorFlow Estimator BestExporter event_file_pattern ничего не делает

0 голосов
/ 21 ноября 2018

У меня есть colab с очень простой демонстрацией Estimator для единственной цели изучения / понимания API Estimator с целью создания соглашения для модели plug-and-play сраспространенные помехи в торговле (например, ранняя остановка, если набор проверки перестает улучшаться, экспорт модели и т. д.).

В ячейке , где я определяю BestExporter У меня есть: 

exporter = tf.estimator.BestExporter(
    name="best_exporter",
    serving_input_receiver_fn=serving_input_receiver_fn,
    # event_file_pattern="model_*", # <--- doesn't do anything?
    exports_to_keep=5
) # this will keep the 5 best checkpoints

, которая в настоящее время экспортирует мои лучшие модели в 

<MODEL_DIR>/export/best_exporter/<some_number>

где MODEL_DIR=./test (под ячейкой setup > constants и указано в RunConfig, установленном в estimator functions > run config)

Если я раскомментирую event_file_pattern="model_*",, я ожидал увидеть, что моя модель будет экспортированадо <MODEL_DIR>/export/best_exporter/model_<number>, но это не так.

Так как мне указать, как должны называться мои модели?

Например, в colab я отслеживаю метрики mse и mae.

Как мне получить 5 лучших exports_to_keep сохранить как: 

MODEL_DIR/export/best_exporter/<model_name>_mae_<mae-rounded-to-3-decimals>

?

...