From e051c36c4cd4e1d903472ae441378cba88949082 Mon Sep 17 00:00:00 2001
From: Readm <46922107+readmlll@users.noreply.github.com>
Date: Fri, 8 Sep 2023 10:36:30 +0800
Subject: [PATCH] Update build_datasets.py

Floating-point numbers cause errors in the assertions below
---
 model/build_datasets.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/model/build_datasets.py b/model/build_datasets.py
index 513d883..750ed36 100755
--- a/model/build_datasets.py
+++ b/model/build_datasets.py
@@ -37,6 +37,9 @@
 evaluation_samples_number = len(paths) / (distribution + 1)
 training_samples_number = evaluation_samples_number * distribution
 
+evaluation_samples_number=int(evaluation_samples_number+0.5)
+training_samples_number=int(training_samples_number+0.5)
+
 assert training_samples_number + evaluation_samples_number == len(paths)
 
 print("Splitting datasets, training samples: {}, evaluation samples: {}".format(training_samples_number, evaluation_samples_number))