Skip to content

Commit 739069b

Browse files
committed
hotfix
1 parent de170f3 commit 739069b

File tree

3 files changed

+14
-3
lines changed

3 files changed

+14
-3
lines changed

docs/src/man/preprocess.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,10 @@ CurrentModule = JudiLing
55
# Preprocess
66

77
```@docs
8-
train_val_split
98
lpo_cv_split
109
loo_cv_split
1110
lpo_cv_split(p, data_path)
1211
loo_cv_split(data_path)
12+
train_val_random_split(data_path, output_dir_path, data_prefix)
13+
train_val_carefully_split(data_path, output_dir_path, data_prefix, n_features_columns)
1314
```

src/preprocess.jl

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,11 @@ function loo_cv_split(data_path; random_seed = 314)
3737
lpo_cv_split(1, data_path)
3838
end
3939

40+
"""
41+
train_val_random_split(data_path, output_dir_path, data_prefix)
42+
43+
Randomly split dataset.
44+
"""
4045
function train_val_random_split(
4146
data_path,
4247
output_dir_path,
@@ -74,11 +79,16 @@ function train_val_random_split(
7479
nothing
7580
end
7681

82+
"""
83+
train_val_carefully_split(data_path, output_dir_path, data_prefix, n_features_columns)
84+
85+
Carefully split dataset.
86+
"""
7787
function train_val_carefully_split(
7888
data_path,
7989
output_dir_path,
90+
data_prefix,
8091
n_features_columns;
81-
data_prefix = "data",
8292
train_sample_size = 0,
8393
val_sample_size = 0,
8494
val_ratio = 0.0,

src/test_combo.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -763,8 +763,8 @@ function loading_data_carefully_split(
763763
train_val_carefully_split(
764764
data_path,
765765
output_dir_path,
766+
data_prefix,
766767
n_features_columns,
767-
data_prefix = data_prefix,
768768
train_sample_size = train_sample_size,
769769
val_sample_size = val_sample_size,
770770
val_ratio = val_ratio,

0 commit comments

Comments
 (0)