keras常见层用法备忘

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

猜你喜欢

转载自www.cnblogs.com/databingo/p/9222052.html
今日推荐