矩阵可以作为Keras嵌入层的输入吗?

时间:2018-11-05 10:06:05

标签: python machine-learning keras nlp embedding

我正在使用Keras捕获数据集的语义信息。而且我已经将数据标记为整数向量。它具有这样的形式:

texts=[[1,2,3,2,1],
       [2,3,4,2,2],
       [3,33,2,1,3]]

labels=[1,0,1]

并且标签仅包含0或1,每个列表包含一个标签。 我想使用Keras的嵌入层来嵌入它。但是Internet上的示例仅包含一个列表:

texts=[1,2,3,4,2,1]

我想知道我可以将矩阵输入到嵌入层吗?

1 个答案:

答案 0 :(得分:1)

endtickets列表中的每个列表都是一个训练样本,并且usertickets列表中的每个列表都有相应的标签。因此,每个训练样本都是整数(即单词索引)的向量,您可以轻松地将其输入到嵌入层:

user_id

请注意,您可能需要将训练数据和标签转换为numpy数组(如果尚未转换):

texts