Печать значения Numeri c из словаря в файл Yaml

Question

Я отслеживаю модель Tensorflow, и у меня настроены показатели для измерения потери пакетов и потерь эпох, которые я ввожу в словарь, используя этот код python:

benchmark_data = dict()  
benchmark_data["batch_loss"] = batch_loss[-1]  
benchmark_data["epoch_loss"] = epoch_loss[-1]

Затем я пытаюсь вывести дамп эти данные в файл yaml с:

with open(output_filename, "w") as output_file:  
     yaml.dump(benchmark_data, output_file,Dumper=yaml.Dumper)

Но файл yaml печатается как:

batch_loss: !!python/object/apply:numpy.core.multiarray.scalar
- !!python/object/apply:numpy.dtype
  args:
  - f4
  - 0
  - 1
  state: !!python/tuple
  - 3
  - <
  - null
  - null
  - null
  - -1
  - -1
  - 0
- !!binary |
  /Um+PQ==

epoch_loss: !!python/object/apply:numpy.core.multiarray.scalar
- !!python/object/apply:numpy.dtype
  args:
  - f8
  - 0
  - 1
  state: !!python/tuple
  - 3
  - <
  - null
  - null
  - null
  - -1
  - -1
  - 0
- !!binary |
  foMQ3oiGwz8=

Как получить их для печати в виде простого числа c? Любая помощь будет принята с благодарностью.

Спасибо