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

Test with dataset bigger than 100000

parent dd59a606
No related branches found
No related tags found
1 merge request!2Correct bug
Pipeline #3201 passed
This commit is part of merge request !2. Comments created here will be created in the context of that merge request.
...@@ -32,5 +32,20 @@ public final class Main { ...@@ -32,5 +32,20 @@ public final class Main {
System.out.println("PR AUC with trapezoid : " + pr.computePRAUC()); System.out.println("PR AUC with trapezoid : " + pr.computePRAUC());
pr.computePrcPointsAndGenerateCurve("PR_curve.png"); pr.computePrcPointsAndGenerateCurve("PR_curve.png");
double[] long_data = new double[200000];
double[] long_alert = new double[long_data.length];
for (int i = 0; i < long_data.length; i++) {
long_data[i] = Math.random();
double trig = Math.random();
if (trig < 0.2) {
long_alert[i] = 0.0;
} else {
long_alert[i] = 1.0;
}
}
Roc long_roc = new Roc(long_data, long_alert);
System.out.println("Long AUC : " + long_roc.computeAUC());
long_roc.computeRocPointsAndGenerateCurve("long_roc.png");
} }
} }
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