Слой Keras неизвестен, попробуйте загрузить модель

Question

Я сохранил обученную модель с этим кодом, но не могу загрузить ее, потому что слой не находится в keras.layer, вот мой код, заранее спасибо за вашу драгоценную помощь!

from __future__ import absolute_import, division, print_function, unicode_literals

import numpy as np

import tensorflow as tf
import tensorflow_hub as hub
import tensorflow_datasets as tfds

train_data, validation_data, test_data = tfds.load(
    name="imdb_reviews", 
    split=[
   tfds.Split.TRAIN.subsplit(tfds.percent[:60]),
   tfds.Split.TRAIN.subsplit(tfds.percent[60:]),
   'test'
],
    as_supervised=True)

train_examples_batch, train_labels_batch = next(iter(train_data.batch(10)))

# to use for more accuracy... : google/tf2-preview/nnlm-en-dim128/1
embedding = "https://tfhub.dev/google/tf2-preview/gnews-swivel-20dim/1"
hub_layer = hub.KerasLayer(embedding, input_shape=[], 
                           dtype=tf.string, trainable=True)
hub_layer(train_examples_batch[:3])

model = tf.keras.Sequential()
model.add(hub_layer)
model.add(tf.keras.layers.Dense(16, activation='relu'))
model.add(tf.keras.layers.Dense(1))

model.summary()

model.compile(optimizer='adam',
              loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=['accuracy'])


history = model.fit(train_data.shuffle(10000).batch(512),
                    epochs=20,
                    validation_data=validation_data.batch(512),
                    verbose=1)

results = model.evaluate(test_data.batch(512), verbose=1)

for name, value in zip(model.metrics_names, results):
  print("%s: %.3f" % (name, value))

# serialize model to JSON
model_json = model.to_json()
with open("my_model.json", "w") as json_file:
    json_file.write(model_json)
# serialize weights to HDF5
model.save_weights("w_model.h5")

, когда я пытаюсь чтобы загрузить его, я получил проблему, потому что он не распознает слой "hub_layer", созданный с помощью набора данных tenorflow. Надеюсь, ты знаешь, как с этим справиться.

Answer 1

проблема здесь в том, что файл сохранения модели не содержит код этого слоя. Вы можете решить эту проблему, используя custom_objects и передать словарь этому аргументу.

Хитрость заключается в том, чтобы скопировать нераспознанное имя, поместить его в строку в словаре в качестве ключа и указать этот слой в качестве значения

Например

import tensorflow_hub as hub
layer_dict = {
    "hub_layer":hub.KerasLayer
}

model = model_from_json('model_file',custom_objects=layer_dict)