Skip to content

Commit

Permalink
mend
Browse files Browse the repository at this point in the history
  • Loading branch information
amva13 committed Oct 7, 2024
1 parent fcce378 commit 0262f79
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions tdc/utils/split.py
Original file line number Diff line number Diff line change
Expand Up @@ -351,8 +351,8 @@ def create_group_split(train_val, seed, holdout_frac, group_column):
train_val_temp = train_val[train_val[group_column] == i]
np.random.seed(seed)
msk = np.random.rand(len(train_val_temp)) < (1 - holdout_frac)
train_df = pd.concat([train_df,train_val_temp[msk]], axis=0)
val_df = pd.concat([val_df,train_val_temp[~msk]], axis=0)
train_df = pd.concat([train_df, train_val_temp[msk]], axis=0)
val_df = pd.concat([val_df, train_val_temp[~msk]], axis=0)

return {
"train": train_df.reset_index(drop=True),
Expand Down
2 changes: 1 addition & 1 deletion tdc/version.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,4 +19,4 @@
# Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
# 'X.Y.dev0' is the canonical version of 'X.Y.dev'
#
__version__ = "1.0.7" # pragma: no cover
__version__ = "1.1.0" # pragma: no cover

0 comments on commit 0262f79

Please sign in to comment.