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

Add setters in HyperParemeters class

parent f1e30dbf
No related branches found
No related tags found
1 merge request!4Neural network
Pipeline #2268 failed
...@@ -27,11 +27,11 @@ public class HyperParameters { ...@@ -27,11 +27,11 @@ public class HyperParameters {
final OptimizationAlgorithm algorithm, final OptimizationAlgorithm algorithm,
final Activation activation_function, final Activation activation_function,
final double percent_test_train) { final double percent_test_train) {
this.neurons_number = neurons_number; setNeuronsNumber(neurons_number);
this.learning_rate = learning_rate; setLearningRate(learning_rate);
this.algorithm = algorithm; setAlgorithm(algorithm);
this.activation_function = activation_function; setActivationFunction(activation_function);
this.percent_test_train = percent_test_train; setPercentTestTrain(percent_test_train);
} }
/** /**
...@@ -47,7 +47,7 @@ public class HyperParameters { ...@@ -47,7 +47,7 @@ public class HyperParameters {
final OptimizationAlgorithm algorithm, final OptimizationAlgorithm algorithm,
final Activation activation_function) { final Activation activation_function) {
this(neurons_number, learning_rate, algorithm, this(neurons_number, learning_rate, algorithm,
activation_function, 100); activation_function, 99);
} }
/** /**
...@@ -89,4 +89,51 @@ public class HyperParameters { ...@@ -89,4 +89,51 @@ public class HyperParameters {
public final double getPercentTestTrain() { public final double getPercentTestTrain() {
return percent_test_train; return percent_test_train;
} }
/**
* @param neurons_number
*/
public void setNeuronsNumber(final int neurons_number) {
if (neurons_number < 5 || neurons_number > 80) {
throw new IllegalArgumentException(
"Neuron number must be between 5 and 80");
}
this.neurons_number = neurons_number;
}
/**
* @param learning_rate
*/
public void setLearningRate(final double learning_rate) {
if (learning_rate <= 0.0 || learning_rate >= 1.0) {
throw new IllegalArgumentException(
"Learning rate must be between 0 and 1");
}
this.learning_rate = learning_rate;
}
/**
* @param algorithm
*/
public void setAlgorithm(final OptimizationAlgorithm algorithm) {
this.algorithm = algorithm;
}
/**
* @param activation_function
*/
public void setActivationFunction(final Activation activation_function) {
this.activation_function = activation_function;
}
/**
* @param percent_test_train
*/
public void setPercentTestTrain(final double percent_test_train) {
if (percent_test_train <= 0 || percent_test_train >= 100) {
throw new IllegalArgumentException(
"Percentage of train must be between 0 and 100");
}
this.percent_test_train = percent_test_train;
}
} }
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