Извлечь структуру данных типа графа из tf.keras.model - PullRequest
Новая
       11

Извлечь структуру данных типа графа из tf.keras.model

0 голосов
/ 04 марта 2020

Я хочу извлечь граф в python формате данных из tf.keras.model.

Я пытаюсь получить структуру данных графа из модели DNN в виде tf.keras. .модель. 

model = keras.Sequential([
keras.layers.Flatten(input_shape=(28, 28)),
keras.layers.Dense(128, activation='relu'),
keras.layers.Dense(10)])

Модель выглядит выше. Я могу получить график в TensorBoard по коду ниже 

logdir="logs/fit/" + datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = keras.callbacks.TensorBoard(log_dir=logdir)
model.fit(train_images, train_labels, epochs=10, callbacks=[tensorboard_callback])

График похож на ниже

enter image description here

Когда я нажимаю один из узел, я мог бы получить атрибуты этого узла в python формате данных.

То, что я хочу получить, это график в формате данных python, а не в визуализированном графике.

Как можно извлечь узлы графа и его атрибуты в текстовом формате (или python формат данных) из tf.keras.model?

...