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

Correction in main class

parent 1f03525d
No related branches found
No related tags found
1 merge request!3K fold
Pipeline #1764 passed
...@@ -73,17 +73,14 @@ public final class Example { ...@@ -73,17 +73,14 @@ public final class Example {
+ (end_time - start_time) / 1000 + " seconds"); + (end_time - start_time) / 1000 + " seconds");
System.out.println("Run Cross validation"); System.out.println("Run Cross validation");
List<AbstractSolution> solutions = trainer.runKFold( List<Double> solutions = trainer.runKFold(
data_file, data_file,
expected_file, expected_file,
10); 10);
for (int i = 0; i < solutions.size(); i++) { for (int i = 0; i < solutions.size(); i++) {
System.out.println(solutions.get(i)); System.out.println(solutions.get(i));
System.out.println("AUC Value for fold" System.out.println("AUC Value for fold" + i + solutions);
+ i + solutions.get(i).computeAUC(
data_file,
expected_file));
} }
} }
......
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