search

seyonechithrananda / bert-loves-chemistry

bert-loves-chemistry: a repository of HuggingFace models applied on chemical SMILES data for drug design, chemical modelling, etc.
MIT License
389 stars 60 forks source link

Csv finetune #37

Closed ElanaPearl closed 3 years ago

ElanaPearl commented 3 years ago

modified finetune script so it can finetune on non molnet, csv datasets

If you pass in a directory, it should have the structure dataset_name/ -train.csv -valid.csv -test.csv