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)
using Random
n = size(X, 1);
nTrainValTest = zeros(3);
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)
using Random
n = X;
nTrainValTest = zeros(3);
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)
using Random
n = length(G);
splits = convert(Matrix{Int64}, zeros(n, nSplits));