@@ -36,7 +36,6 @@ class XYBaseDataModule(LightningDataModule):
3636 label_filter (Optional[int]): The index of the label to filter. Default is None.
3737 balance_after_filter (Optional[float]): The ratio of negative samples to positive samples after filtering. Default is None.
3838 num_workers (int): The number of worker processes for data loading. Default is 1.
39- chebi_version (int): The version of ChEBI to use. Default is 200.
4039 inner_k_folds (int): The number of folds for inner cross-validation. Use -1 to disable inner cross-validation. Default is -1.
4140 fold_index (Optional[int]): The index of the fold to use for training and validation. Default is None.
4241 base_dir (Optional[str]): The base directory for storing processed and raw data. Default is None.
@@ -52,7 +51,6 @@ class XYBaseDataModule(LightningDataModule):
5251 label_filter (Optional[int]): The index of the label to filter.
5352 balance_after_filter (Optional[float]): The ratio of negative samples to positive samples after filtering.
5453 num_workers (int): The number of worker processes for data loading.
55- chebi_version (int): The version of ChEBI to use.
5654 inner_k_folds (int): The number of folds for inner cross-validation. If it is less than to, no cross-validation will be performed.
5755 fold_index (Optional[int]): The index of the fold to use for training and validation (only relevant for cross-validation).
5856 _base_dir (Optional[str]): The base directory for storing processed and raw data.
@@ -75,7 +73,6 @@ def __init__(
7573 label_filter : Optional [int ] = None ,
7674 balance_after_filter : Optional [float ] = None ,
7775 num_workers : int = 1 ,
78- chebi_version : int = 200 ,
7976 inner_k_folds : int = - 1 , # use inner cross-validation if > 1
8077 fold_index : Optional [int ] = None ,
8178 base_dir : Optional [str ] = None ,
@@ -95,7 +92,6 @@ def __init__(
9592 ), "Filter balancing requires a filter"
9693 self .balance_after_filter = balance_after_filter
9794 self .num_workers = num_workers
98- self .chebi_version = chebi_version
9995 assert type (inner_k_folds ) is int
10096 self .inner_k_folds = inner_k_folds
10197 self .use_inner_cross_validation = (
0 commit comments