Improve Trainer
and DeeprankDataset
for production testing
#510
Labels
stale
issue not touched from too much time
Trainer
and DeeprankDataset
for production testing
#510
There are some issues when using the package for testing a pre-trained model on newly generated data:
GraphDataset
class requiresdataset_train
as input even in such cases (whenevertrain
is False). We should be able of using a test dataset without the need of the original model's training dataset. We can use the info stored in the pre-trained model for inherit the needed attributes. (see_check_inherited_params
in dataset.py)Trainer
class' init, before loading parameters and the pretrained model there is a check for the target, which in a pre-trained model case could be not present at all.Trainer
class expects the attributeepoch_saved_model
, which should be saved within the state of the pre-trained model.ValueError("All arrays must be of the same length")
)In order to make reasonable changes, I think we need to take into account all the possible scenarios using a mock example:
The text was updated successfully, but these errors were encountered: