keras 中 Embedding层input_dim,output_dim个人理解

本文深入解析了Keras中的Embedding层,解释了如何通过该层将单词索引转换为向量表示,介绍了input_dim和output_dim参数的含义及使用方法。input_dim参数代表词汇表大小,即最大整数索引+1;output_dim则定义了密集向量的维度。

from keras.layers.embeddings import Embedding
Embedding(input_dim, output_dim, embeddings_initializer='uniform', embeddings_regularizer=None, activity_regularizer=None, embeddings_constraint=None, mask_zero=False, input_length=None)

  • input_dim: int > 0. Size of the vocabulary, i.e. maximum integer index + 1.
  • output_dim: int >= 0. Dimension of the dense embedding.
  • embeddings_initializer: Initializer for the embeddings matrix (see initializers).

上面只是列出了常用参数,其中input_dim便是词汇长度+1,我个人理解是,Embedding功能是将单词索引变为向量表示,这个过程,便是一个降维的过程,所以你的输入维度便是你的最大数字加一
在这里插入图片描述
可以看出,我输入数组中最大数字是3,所以我的input_dim便要比3要大1,其中output_dim便是输出维度,就如同CNN最后的全连接层一样,上面我设置的5,便将每一位要表示的数字变为用1x5来表示的向量。

ValueError Traceback (most recent call last) Cell In[21], line 15 13 return model 14 # 调用 ---> 15 model = build_model(vocab_size=len(vocab), embedding_dim=embedding_dim, rnn_units=rnn_units, batch_size=BATCH_SIZE) 16 model.summary() Cell In[21], line 9, in build_model(vocab_size, embedding_dim, rnn_units, batch_size) 7 def build_model(vocab_size, embedding_dim, rnn_units, batch_size): 8 model = keras.Sequential([ ----> 9 keras.layers.Embedding(vocab_size, embedding_dim,batch_input_shape=[batch_size, None]), 10 keras.layers.GRU(rnn_units, return_sequences=True, stateful=True, recurrent_initializer='glorot_uniform'), 11 keras.layers.Dense(vocab_size) 12 ]) 13 return model File D:\Tools\Anaconda\Lib\site-packages\keras\src\layers\core\embedding.py:93, in Embedding.__init__(self, input_dim, output_dim, embeddings_initializer, embeddings_regularizer, embeddings_constraint, mask_zero, weights, lora_rank, **kwargs) 89 if input_length is not None: 90 warnings.warn( 91 "Argument `input_length` is deprecated. Just remove it." 92 ) ---> 93 super().__init__(**kwargs) 94 self.input_dim = input_dim 95 self.output_dim = output_dim File D:\Tools\Anaconda\Lib\site-packages\keras\src\layers\layer.py:289, in Layer.__init__(self, activity_regularizer, trainable, dtype, autocast, name, **kwargs) 287 self._input_shape_arg = input_shape_arg 288 if kwargs: --> 289 raise ValueError( 290 "Unrecognized keyword arguments " 291 f"passed to {self.__class__.__name__}: {kwargs}" 292 ) 294 self._path = None # Will be determined in `build_wrapper` 295 self.built = False ValueError: Unrecognized keyword arguments passed to Embedding: {'batch_input_shape': [64, None]}
04-29
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值