New attempt...

This commit is contained in:
Joakim Skogholt 2023-05-10 13:50:37 +02:00
parent f8f6640484
commit 6074e56de2

View file

@ -13,7 +13,7 @@ XVal = my_split["XVal"];
=# =#
using Random
@ -36,8 +36,6 @@ 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));
@ -84,8 +82,6 @@ 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));
@ -160,8 +156,6 @@ 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));