ggfm.data.generate_lm_embs¶

class ggfm.data.generate_lm_embs(all_name, tokenizer, model, indxs, device)[source]¶

Bases:

Language model generates node embeddings for words。

Parameters:

all_name (list) – Words list.
tokenizer – LM tokenizer.
model – Language model.
indxs (list) – Index positions of generated node embeddings.
device (int) – Device.