hoon9405/UniHPF

Many missing attributes in training code

Opened this issue · 1 comments

Many required attributes by file mlm_trainer.py are missing in training main.py
For example: split_method, column_embed, bert_model, etc.
Can you please cross-check and suggest the values of those?
@hoon9405 @starmpcc

Additionally many errors are there in base_dataset.py and other files, it would be really helpful if you could check the uploaded code files once.

To list few of them:

  • missing UnifiedEHRDataset defintion
  • icu_type, split_method arguments missing

Thanks!