We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 14a36e4 commit 7d977d6Copy full SHA for 7d977d6
1 file changed
src/_data_splitting/indexing.jl
@@ -62,9 +62,9 @@ function get_split_indices(data_size::Int;
62
val_end = min(val_end, data_size)
63
64
# Create splits
65
- train_indices = indices[1:train_end]
66
- val_indices = indices[train_end+1:val_end]
67
- test_indices = indices[val_end+1:end]
+ train_indices = @view indices[1:train_end]
+ val_indices = @view indices[train_end+1:val_end]
+ test_indices = @view indices[val_end+1:end]
68
69
return (train=train_indices, val=val_indices, test=test_indices)
70
end
0 commit comments