tf.compat.v1.keras.estimator.model_to_estimator

Создаёт экземпляр Estimator на основе заданной модели Keras.

tf.compat.v1.keras.estimator.model_to_estimator(
    keras_model=None, keras_model_path=None, custom_objects=None, model_dir=None,
    config=None, checkpoint_format='saver'
)

Если вы используете инфраструктуру или инструменты, которые полагаются на Estimators, вы всё равно можете создать модель Keras и использовать model_to_estimator для преобразования модели Keras в Estimator для использования в системах более низкого уровня.

Пример использования см. на странице: Создание estimators из Keras-моделей.

Веса выборок:

Estimators, возвращаемые model_to_estimator, настроены так, чтобы они могли обрабатывать веса выборок (аналогично keras_model.fit(x, y, sample_weights)).

Чтобы передать веса выборок во время обучения или оценки Estimator, первый элемент, возвращаемый функцией ввода, должен быть словарем с ключами features и sample_weights. Пример ниже:

keras_model = tf.keras.Model(...)
keras_model.compile(...)

estimator = tf.keras.estimator.model_to_estimator(keras_model)

def input_fn():
  return dataset_ops.Dataset.from_tensors(
      ({'features': features, 'sample_weights': sample_weights},
       targets))

estimator.train(input_fn, steps=1)