Как настроить вход модели подкласса keras в тензор потока? - PullRequest
Новая
       10

Как настроить вход модели подкласса keras в тензор потока?

0 голосов
/ 29 сентября 2019

Я создал модель подкласса keras, используя тензор потока.Фрагменты приведены ниже.

class SubModel(Model):
    def call(self, inputs):
        print(inputs)

model = SubModel()
model.fit(data, labels, ...)

Когда fit модель, она получит входные данные и сам input_shape.Что я хочу сделать, так это передать входные данные в модель. Так же, как это делает функциональный API. 

inputs = tf.keras.input(shape=(100,))
model = tf.keras.Model(inputs=inputs, outputs=outputs)

1 Ответ

1 голос
/ 29 сентября 2019

Что-то в этом роде? 

model_ = SubModel()
inputs = tf.keras.input(shape=(100,))
outputs = model_(inputs)
model = tf.keras.Model(inputs=inputs, outputs=outputs)
...