From 52af6f00ec757d0271fdff320837abfe12ba6402 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 14 Nov 2024 22:03:22 +0000 Subject: [PATCH] dlr/dataset_mono: fix crash in new ssplit3 setup --- aimodel/src/lib/dataset/dataset_mono.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_mono.py b/aimodel/src/lib/dataset/dataset_mono.py index 7e5e8b3..1a428be 100644 --- a/aimodel/src/lib/dataset/dataset_mono.py +++ b/aimodel/src/lib/dataset/dataset_mono.py @@ -165,7 +165,7 @@ def dataset_mono(dirpath_input, percentage_validate=0.2, percentage_test=0, **kw filepaths_count = len(filepaths) split_trainvalidate=math.floor(filepaths_count * (1-(percentage_validate+percentage_test))) - split_validatetest=math.floor(filepaths * (1 - percentage_test)) + split_validatetest=math.floor(filepaths_count * (1 - percentage_test)) filepaths_train = filepaths[:split_trainvalidate]