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

Solve checkstyle issues

parent 44cbeb73
No related branches found
No related tags found
1 merge request!4Neural network
Pipeline #2335 passed
...@@ -65,13 +65,15 @@ public final class MainTest { ...@@ -65,13 +65,15 @@ public final class MainTest {
System.out.println("Wowa training"); System.out.println("Wowa training");
HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10); HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10);
long end_time = System.currentTimeMillis(); long end_time = System.currentTimeMillis();
System.out.println("Execution time : " + (end_time - start_time) / 1000 + " seconds"); System.out.println("Execution time : " + (end_time - start_time) / 1000
+ " seconds");
start_time = System.currentTimeMillis(); start_time = System.currentTimeMillis();
System.out.println("Neural Network learning"); System.out.println("Neural Network learning");
HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10); HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10);
end_time = System.currentTimeMillis(); end_time = System.currentTimeMillis();
System.out.println("Execution time : " + (end_time - start_time) / 1000 + " seconds"); System.out.println("Execution time : " + (end_time - start_time) / 1000
+ " seconds");
double nn_score = 0.0; double nn_score = 0.0;
double wt_score = 0.0; double wt_score = 0.0;
......
...@@ -138,7 +138,7 @@ public final class NeuralNetwork { ...@@ -138,7 +138,7 @@ public final class NeuralNetwork {
record_reader, 12468, FEATURES_COUNT, CLASSES_COUNT); record_reader, 12468, FEATURES_COUNT, CLASSES_COUNT);
DataSet all_data = iterator.next(); DataSet all_data = iterator.next();
//DataSet training_data //DataSet training_data
// = prepareDataSetForTrainingAndTesting(all_data).getTrain(); // = prepareDataSetForTrainingAndTesting(all_data).getTrain();
return learning(all_data); return learning(all_data);
} catch (IOException e) { } catch (IOException e) {
......
...@@ -162,7 +162,7 @@ public class Trainer { ...@@ -162,7 +162,7 @@ public class Trainer {
* @param increase_ratio * @param increase_ratio
* @return * @return
*/ */
HashMap<AbstractSolution, Double> runKFold( final HashMap<AbstractSolution, Double> runKFold(
final List<TrainingDataset> prepared_folds, final List<TrainingDataset> prepared_folds,
final int increase_ratio) { final int increase_ratio) {
HashMap<AbstractSolution, Double> map = new HashMap<>(); HashMap<AbstractSolution, Double> map = new HashMap<>();
......
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