Trying to fix problems with Random

This commit is contained in:
Joakim Skogholt 2023-05-10 13:46:28 +02:00
parent 6394463206
commit f8f6640484

View file

@ -36,6 +36,8 @@ Creates training/validation/test split for dataset
""" """
function createDataSplitInds(X, nSplits, props=[6/10, 2/10, 2/10], rngseed=42) function createDataSplitInds(X, nSplits, props=[6/10, 2/10, 2/10], rngseed=42)
using Random
n = size(X, 1); n = size(X, 1);
nTrainValTest = zeros(3); nTrainValTest = zeros(3);
splits = convert(Matrix{Int64}, zeros(n, nSplits)); splits = convert(Matrix{Int64}, zeros(n, nSplits));
@ -82,6 +84,8 @@ Creates training/validation/test split for dataset
""" """
function createDataSplitInds(X::Int64, nSplits, props=[6/10, 2/10, 2/10], rngseed=42) function createDataSplitInds(X::Int64, nSplits, props=[6/10, 2/10, 2/10], rngseed=42)
using Random
n = X; n = X;
nTrainValTest = zeros(3); nTrainValTest = zeros(3);
splits = convert(Matrix{Int64}, zeros(n, nSplits)); splits = convert(Matrix{Int64}, zeros(n, nSplits));
@ -156,6 +160,8 @@ Creates stratified training/validation/test split for dataset. Assumes G is code
""" """
function createDataSplitBinaryStratified(G::Vector, nSplits, props=[6/10, 2/10, 2/10], rngseed=42) function createDataSplitBinaryStratified(G::Vector, nSplits, props=[6/10, 2/10, 2/10], rngseed=42)
using Random
n = length(G); n = length(G);
splits = convert(Matrix{Int64}, zeros(n, nSplits)); splits = convert(Matrix{Int64}, zeros(n, nSplits));