about word_embedding_c of WordKVMN

[Alexia1994]

Excuse me, why self.word_embedding_c = nn.Embedding(10, hidden_size)? I do not really understand why is 10.
Thanks for your answering : )

[yuanheTian]

Excuse me, why self.word_embedding_c = nn.Embedding(10, hidden_size)? I do not really understand why is 10.
Thanks for your answering : )

Thanks for asking!

That number actually should be the size of the value set. We map the value to its corresponding index in line 353 of wmseg_model.py. The range of the value index should be [0, 4]. So, using 10 here will be enough to map the index to its embedding.