Извлечение информации из схемы метаданных преобразования тензорного потока

Question

I сохранена модель конвейера преобразования модель (артефакты, tfrecords, ...). Функции входных данных содержат словарь (созданный с помощью tft.compute_and_apply_vocabulary).
После прочтения сохраненных данных мне теперь нужна информация о словаре для создания моей model_fn.

Я нашел способ сделать это, но подумал, есть ли более чистый подход. Вот мой код:

transformed_metadata = metadata_io.read_metadata(os.path.join(ARTIFACT_DIR, "transformed_metadata"))
tfdv.get_domain(transformed_metadata.schema.__dict__['_schema_proto'], 'text')

, что дает (по желанию):

мин: -1 макс: 21424 is_categorical: true

schema.pbtxt выглядит так:

feature {
  name: "text"
  type: INT
  int_domain {
    min: -1
    max: 21424
    is_categorical: true
  }
}
feature {
  name: "updown"
  type: INT
  presence {
    min_fraction: 1.0
  }
  shape {
    dim {
      size: 2
    }
  }
}