Thank you for this great project! I want to modify the pretrain model and find that one of the input v_token seems missing, referred in pretrain_model.py as:
if not self.pretrain:
return src
cell_embs.extend(emb)
v_token = data["v_token"]
But the dataset seems not providing this token series? in HCADataset.get_sample:
return {
"u_token": u_token,
"ds": ds[u_cols],
"ds_norm": ds_norm[u_cols],
"perturb": np.array([perturb]),
"mask_attn": self.reader.get_mask(u_cols),
"index": u_cols
}
Thanks in advance for clarifing!
Thank you for this great project! I want to modify the pretrain model and find that one of the input
v_tokenseems missing, referred inpretrain_model.pyas:But the dataset seems not providing this token series? in
HCADataset.get_sample:Thanks in advance for clarifing!