From 194b6827506ca1e31f8f5ad5129e8fbc430a4864 Mon Sep 17 00:00:00 2001
From: Alex <croix.alexandre@gmail.com>
Date: Wed, 14 Aug 2019 10:37:24 +0200
Subject: [PATCH] Correction in prepareFolds method. Issue was an error in the
number of true_aler elements in each folds
---
src/main/java/be/cylab/java/wowa/training/Trainer.java | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/main/java/be/cylab/java/wowa/training/Trainer.java b/src/main/java/be/cylab/java/wowa/training/Trainer.java
index dd394b9..b1db1d9 100644
--- a/src/main/java/be/cylab/java/wowa/training/Trainer.java
+++ b/src/main/java/be/cylab/java/wowa/training/Trainer.java
@@ -188,11 +188,11 @@ public class Trainer {
List<Double> expected = dataset.getExpected();
List<TrainingDataset> fold_dataset = new ArrayList<>();
//Check if it is rounded !!!!
- int element_number_in_fold = expected.size() / fold_number;
int alert_number
= (int) Math.floor(Utils.sumListElements(expected)
/ fold_number);
- int no_alert_number = element_number_in_fold - alert_number;
+ int no_alert_number = (int) (expected.size()
+ - Utils.sumListElements(expected)) / fold_number;
for (int i = 0; i < fold_number; i++) {
TrainingDataset tmp = new TrainingDataset();
--
GitLab