Проблема с встраиванием предварительно обученной модели в Керас

Question

У меня есть предварительно обученная модель Fasttext, и я хочу встроить ее в Keras.

model = Sequential()
model.add(Embedding(MAX_NB_WORDS, 
                    EMBEDDING_DIM, 
                    input_length=X.shape[1],
                    input_length=4,
                    weights=[embedding_matrix],
                    trainable=False))

Но это не сработало.

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

Кажется, что weights и embeddings_initializer устарели.

Есть ли какой-нибудь альтернативный метод для решения проблемы? Заранее спасибо

Answer 1

Параметр Weights устарел в слое Embedded Keras.

Новая версия встраиваемого слоя будет выглядеть следующим образом -

embedding_layer = Embedding(num_words,
                            EMBEDDING_DIM,
                            embeddings_initializer=Constant(embedding_matrix),
                            input_length=MAX_SEQUENCE_LENGTH,
                            trainable=False)

Вы можете найти последнюю версию встраиваемого слоя здесь - Keras Embedded Layer

Вы можете найти пример встраивания предварительно подготовленного слова здесь - Встраивание предварительно подготовленного слова