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))