Skip to content
Snippets Groups Projects
Commit 194b6827 authored by a.croix's avatar a.croix
Browse files

Correction in prepareFolds method. Issue was an error in the number of...

Correction in prepareFolds method. Issue was an error in the number of true_aler elements in each folds
parent 8e142cf2
No related branches found
No related tags found
No related merge requests found
Pipeline #2121 failed
...@@ -188,11 +188,11 @@ public class Trainer { ...@@ -188,11 +188,11 @@ public class Trainer {
List<Double> expected = dataset.getExpected(); List<Double> expected = dataset.getExpected();
List<TrainingDataset> fold_dataset = new ArrayList<>(); List<TrainingDataset> fold_dataset = new ArrayList<>();
//Check if it is rounded !!!! //Check if it is rounded !!!!
int element_number_in_fold = expected.size() / fold_number;
int alert_number int alert_number
= (int) Math.floor(Utils.sumListElements(expected) = (int) Math.floor(Utils.sumListElements(expected)
/ fold_number); / 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++) { for (int i = 0; i < fold_number; i++) {
TrainingDataset tmp = new TrainingDataset(); TrainingDataset tmp = new TrainingDataset();
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment