Skip to content

shrimonmuke0202/EBM-PICO

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Install Packages

  pip3 install flair
  pip3 install seqeval

Datasets

Please Download Datset from Link and followed the method describe in that link to get the CONLL format. and save the data in the Data directory

Run the following code to train the model

    !python3 train.py --dataset_path Data/ \
--data_train train.txt\
--data_test test.txt\
--data_dev dev.txt\
--output_dir model \
--model_name_or_path michiyasunaga/BioLinkBERT-large \
--layers -1\
--subtoken_pooling first_last\
--hidden_size 256\
--learning_rate 5e-5\
--num_epochs 10 \
--use_crf False

Inference

Run the following scrpit to test the best model

  python3 evaluate.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages