diff --git a/src/main/java/be/cylab/java/wowa/training/HyperParameters.java b/src/main/java/be/cylab/java/wowa/training/HyperParameters.java index 5613f7739e4b3eedf7054c3d68ba255ae0d7431c..32a9084958f85d221c55ef1840e3a1cbc6d0c31d 100644 --- a/src/main/java/be/cylab/java/wowa/training/HyperParameters.java +++ b/src/main/java/be/cylab/java/wowa/training/HyperParameters.java @@ -47,14 +47,14 @@ public class HyperParameters { final OptimizationAlgorithm algorithm, final Activation activation_function) { this(neurons_number, learning_rate, algorithm, - activation_function, (Double) null); + activation_function, 100); } /** * Getter for neuron_number. * @return */ - public int getNeuronsNumber() { + public final int getNeuronsNumber() { return neurons_number; } @@ -62,7 +62,7 @@ public class HyperParameters { * Getter for learning rate. * @return */ - public double getLearningRate() { + public final double getLearningRate() { return learning_rate; } @@ -70,7 +70,7 @@ public class HyperParameters { * Getter for backpropagation algorithm. * @return */ - public OptimizationAlgorithm getAlgorithm() { + public final OptimizationAlgorithm getAlgorithm() { return algorithm; } @@ -78,7 +78,7 @@ public class HyperParameters { * Getter for activation function. * @return */ - public Activation getActivationFunction() { + public final Activation getActivationFunction() { return activation_function; } @@ -86,7 +86,7 @@ public class HyperParameters { * Getter for percent_test_train. * @return */ - public double getPercentTestTrain() { + public final double getPercentTestTrain() { return percent_test_train; } } 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 66adbb3de3714435abe37abb9d359ba575ceae5e..82ab18210c9a19c1db05f3a754da44d2c30c30ff 100644 --- a/src/main/java/be/cylab/java/wowa/training/MainDL4J.java +++ b/src/main/java/be/cylab/java/wowa/training/MainDL4J.java @@ -21,7 +21,7 @@ public final class MainDL4J { */ public static void main(final String[] args) { - String file_name = args[0]; + //String file_name = args[0]; double learning_rate = Double.parseDouble(args[1]); OptimizationAlgorithm optimization_algorithm; Activation activation_function;