Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit b0c750c

Browse files
committed
Shuffle data
1 parent f068908 commit b0c750c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

example/FlowOverCircle/src/FlowOverCircle.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ end
3333

3434
function get_dataloader(; ts::AbstractRange=LinRange(100, 11000, 10000), ratio::Float64=0.95, batchsize=100)
3535
data = gen_data(ts)
36-
data_train, data_test = splitobs((𝐱=data[:, :, :, 1:end-1], 𝐲=data[:, :, :, 2:end]), at=ratio)
36+
data_train, data_test = splitobs(shuffleobs((𝐱=data[:, :, :, 1:end-1], 𝐲=data[:, :, :, 2:end])), at=ratio)
3737

3838
loader_train = DataLoader(data_train, batchsize=batchsize, shuffle=true)
3939
loader_test = DataLoader(data_test, batchsize=batchsize, shuffle=false)

0 commit comments

Comments
 (0)