Я использовал keras для использования предварительно обученных вложений слов, но я не совсем уверен, как это сделать на модели scikit-learn.
Мне нужно сделать это и в sklearn, потому что я использую vecstack
для объединения последовательной модели keras и модели sklearn.
Вот что я сделал для модели keras:
glove_dir = '/home/Documents/Glove'
embeddings_index = {}
f = open(os.path.join(glove_dir, 'glove.6B.200d.txt'), 'r', encoding='utf-8')
for line in f:
values = line.split()
word = values[0]
coefs = np.asarray(values[1:], dtype='float32')
embeddings_index[word] = coefs
f.close()
embedding_dim = 200
embedding_matrix = np.zeros((max_words, embedding_dim))
for word, i in word_index.items():
if i < max_words:
embedding_vector = embeddings_index.get(word)
if embedding_vector is not None:
embedding_matrix[i] = embedding_vector
model = Sequential()
model.add(Embedding(max_words, embedding_dim, input_length=maxlen))
.
.
model.layers[0].set_weights([embedding_matrix])
model.layers[0].trainable = False
model.compile(----)
model.fit(-----)
Я очень новичок в scikit-learn, из того, что я видел, чтобы сделать модель в sklearn, вы делаете:
lr = LogisticRegression()
lr.fit(X_train, y_train)
lr.predict(x_test)
Итак, мой вопрос: как мне использовать предварительно обученную перчатку с этой моделью? куда мне сдать предварительно тренированную перчатку embedding_matrix
Большое спасибо, и я очень ценю вашу помощь.