Я считаю, что выходные строки просто индексируются от 0 до input_dim-1, и любое целое число, сгенерированное вашим LabelEncoder для каждого категориального варианта, будет соответствовать соответствующей строке. Например, если у вас есть 3 категориальные переменные: «яблоко», «груша», «апельсин» и LabelEncoder назначает грушу = 2, апельсин = 0, яблоко = 1, то строка 0 матрицы внедрения будет связана с «оранжевым»", строка 1 будет" яблоком ", а строка 2 будет" грушей ".
Приведенный ниже фрагмент рабочего кода демонстрирует идею. Надеюсь, это поможет.
import numpy as np
from keras import Sequential
from keras.layers import Embedding
model = Sequential()
model.add(Embedding(5, 2))
model.compile('rmsprop', 'mse')
#model.summary()
input_array = np.array([[1, 1, 1],
[1, 2, 2]])
output_array = model.predict(input_array)
print('model.get_weights():')
print(model.get_weights())
print('output_array:')
print(output_array)