Added ablation study with randomly sampled vectors + Started working on LightningDataModule wrapper 4e1d3f6 ribesstefano commited on May 2, 2024
Added softmax in model + Fixed and updated some hparams + Fixed bug in tanimoto distance (it was treated as similarity) bda3015 ribesstefano commited on Apr 29, 2024
Working loading and predicting with pretrained model + Added scaling in "pure" pytorch during forwarding 15216c3 ribesstefano commited on Apr 25, 2024
Refactored experiments + fixed bug in dataset when applying scaling to val and test sets 6a5a99e ribesstefano commited on Apr 23, 2024