From 6d1239ab274eac097afc76e988f0bc0f379c1849 Mon Sep 17 00:00:00 2001
From: Alex <croix.alexandre@gmail.com>
Date: Mon, 16 Sep 2019 16:40:26 +0200
Subject: [PATCH] Change loop for number of neurons
---
src/main/java/be/cylab/java/wowa/training/MainDL4J.java | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/main/java/be/cylab/java/wowa/training/MainDL4J.java b/src/main/java/be/cylab/java/wowa/training/MainDL4J.java
index f9f0fa0..b5be713 100644
--- a/src/main/java/be/cylab/java/wowa/training/MainDL4J.java
+++ b/src/main/java/be/cylab/java/wowa/training/MainDL4J.java
@@ -64,7 +64,7 @@ public final class MainDL4J {
}
for (int neuron_number = begin_neuron_number;
- neuron_number < neuron_number_end; neuron_number++) {
+ neuron_number <= neuron_number_end; neuron_number++) {
double average_auc = 0;
System.out.println("Neuron number : " + neuron_number);
HyperParameters parameters
--
GitLab