Skip to content

Commit d982023

Browse files
authored
Merge pull request #342 from PNNL-CompBio/341-datasetsplit_train_others-fails
fixed arguments in train_other()
2 parents e65634b + 2e05d06 commit d982023

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

coderdata/dataset/dataset.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ def split_train_other(
341341
split_type: Literal[
342342
'mixed-set', 'drug-blind', 'cancer-blind'
343343
]='mixed-set',
344-
ratio: tuple[int, int, int]=(8,2),
344+
ratio: tuple[int, int]=(8,2),
345345
stratify_by: Optional[str]=None,
346346
balance: bool=False,
347347
random_state: Optional[Union[int,RandomState]]=None,
@@ -351,7 +351,7 @@ def split_train_other(
351351
split = split_train_other(
352352
data=self,
353353
split_type=split_type,
354-
ration=ratio,
354+
ratio=ratio,
355355
stratify_by=stratify_by,
356356
balance=balance,
357357
random_state=random_state,
@@ -728,19 +728,19 @@ def split_train_other(
728728
split_type: Literal[
729729
'mixed-set', 'drug-blind', 'cancer-blind'
730730
]='mixed-set',
731-
ratio: tuple[int, int, int]=(8,2),
731+
ratio: tuple[int, int]=(8,2),
732732
stratify_by: Optional[str]=None,
733733
balance: bool=False,
734734
random_state: Optional[Union[int,RandomState]]=None,
735735
**kwargs: dict,
736736
):
737737
train, other = _split_two_way(
738-
data,
739-
split_type,
740-
ratio,
741-
stratify_by,
742-
balance,
743-
random_state,
738+
data=data,
739+
split_type=split_type,
740+
ratio=ratio,
741+
stratify_by=stratify_by,
742+
balance=balance,
743+
random_state=random_state,
744744
**kwargs
745745
)
746746
if stratify_by is not None:

0 commit comments

Comments
 (0)