Как сохранить / загрузить часть модели в TensorFlow 2? - PullRequest
Новая
       12

Как сохранить / загрузить часть модели в TensorFlow 2?

1 голос
/ 18 февраля 2020

Я сделал автоэнкодер в TensorFlow 2. Во-первых, мне нужно обучить модель на наборе данных. После обучения автоэнкодеру я хочу использовать часть энкодера для другого набора данных, сохраняя при этом обученные веса. Есть ли способ сделать это? Я не мог найти один. Любые рекомендации будут полезны. Вот грубый автоэнкодер, который я пытаюсь использовать: 

from tensorflow.keras.layers import Input, Dense
from tensorflow.keras import Model

def Autoencoder():
    # Encoder
    inp = Input(shape=(100,))
    x = Dense(1000)(inp)
    x = Dense(500)(x)
    code = Dense(10)(x)

    # Decoder    
    x = Dense(500)(code)
    x = Dense(1000)(x)
    out = Dense(100)(x)

    model = Model(inp,out)
    return model

autoencoder = Autoencoder()
...