Embedding层
keras.layers.Embedding(input_dim, output_dim, embeddings_initializer=‘uniform‘, embeddings_regularizer=None, activity_regularizer=None, embeddings_constraint=None, mask_zero=False, input_length=None, **kwargs)
参数
- input_dim: 字典长度, 单词的数量
- output_dim: 词向量的长度
- weights: 用于初始化的参数矩阵, 大小为(input_dim, output_dim)
- input_length: 单个序列的长度, 即时间步长
- trainable: 是否可训练. 如果想用已有的参数, 此参数需要设置为
False
原文地址:https://www.cnblogs.com/databingo/p/9222052.html
时间: 2024-12-16 01:09:51