Open ilibarra opened 1 year ago
Regarding the second point, currently:
train_iterative
(in mubind prediction.py) called from fit_model
initiates Multibind based on the passed dataset's class, also setting the datatype
parameter.datatype
param.Solution can be
@johschnee may you please yes/no confirm that PBMDataset is the one to be used internally in prediction.py? Please write and/or DM us if more considerations have to be taken. Thank yo, https://github.com/theislab/mubind/blob/main/mubind/tl/prediction.py#L349
I unfortunately don't have access to the mubind-pipeline
repository. But going by your description alone, I'd be careful about renaming. SelexDataset and PBMDataset have different attributes and these attributes are also used by the multibind model.
If you think the code in mubind-pipeline
is relevant for a more datailed answer, it would be great if you could give me access.
@johschnee access to mubind-pipeline
granted. My bad!
Thank you, this indeed changes my answer. Yes, for PBM-Data the class PBMDataset is the one to be used internally. As Ege said, the datatype
parameter is initialized based on the dataset's class.
self.datatype
is enough for this change.