Как сохранить и загрузить модель tf.estimator.BoostedTreesRegressor в версии Tensorflow = '2.0.0'

Question

Я новичок в tf.estimator.BoostedTreesRegressor .. Вот пример кода, который я использовал для построения модели.

n_batches = 20

est = tf.estimator.BoostedTreesRegressor(feature_columns,
                                           n_batches_per_layer=n_batches , learning_rate=0.001, n_trees=700,
                                            max_depth=13, 
model_dir = "model", tf.config.threading.set_intra_op_parallelism_threads(60))

est.train(train_input_fn, max_steps=10)

Я хочу сохранить модель .. и загрузить последнююмодель для прогнозирования продаж.

Подскажите, пожалуйста, как мне это сделать в TensorFlow версии 2 .. как мне не удалось найти ...

Спасибо

Answer 1

Ваша модель должна быть сохранена в пути model_dir в соответствии с официальной документацией . Укажите реальный путь к каталогу model_dir при создании экземпляра BoostedTreesRegressor.

Кроме того, вы можете сохранить модель, используя метод export_saved_model.

# Saving estimator model
serving_input_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(
  tf.feature_column.make_parse_example_spec(feature_columns))
export_path = estimator.export_saved_model("/dir/path/", serving_input_fn)

Для загрузки сохраненной модели вы можете использовать saved_model.load функционирует следующим образом:

#loading saved model
imported = tf.saved_model.load(export_path)