首頁(yè) 猿問(wèn) 在 Keras 中嵌入預(yù)訓(xùn)練模型的問(wèn)題

在 Keras 中嵌入預(yù)訓(xùn)練模型的問(wèn)題

Python

qq_花開(kāi)花謝_0 2022-09-13 17:38:59

我有一個(gè)預(yù)先訓(xùn)練過(guò)的快速文本模型，我想把它嵌入到Keras中。model = Sequential() model.add(Embedding(MAX_NB_WORDS, EMBEDDING_DIM, input_length=X.shape[1], input_length=4, weights=[embedding_matrix], trainable=False))但它沒(méi)有奏效。我發(fā)現(xiàn)很多人在將預(yù)先訓(xùn)練的模型嵌入Keras時(shí)都有同樣的問(wèn)題，而且他們都沒(méi)有解決方案。它似乎已被棄用。weightsembeddings_initializer有沒(méi)有其他方法可以解決這個(gè)問(wèn)題？提前致謝

查看完整描述

1 回答

SMILET

TA貢獻(xiàn)1796條經(jīng)驗(yàn) 獲得超4個(gè)贊

權(quán)重參數(shù)在 Keras 的嵌入層中已棄用。

新版本的嵌入層將如下所示 -

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

您可以在此處找到最新版本的嵌入層詳細(xì)信息 - Keras 嵌入層

您可以在此處找到預(yù)訓(xùn)練詞嵌入的示例 - 預(yù)訓(xùn)練詞嵌入

反對(duì) 回復(fù) 2022-09-13