Skip to content

Commit

Permalink
Move custom threshold logic from make metrics to another PR
Browse files Browse the repository at this point in the history
  • Loading branch information
maurever committed Sep 20, 2023
1 parent 9b74707 commit b490cfa
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,7 @@ public static void gbm_example_flow() {
null,
false, false, -1, null,
false, false, false, false, null,
false, false, false, null, -1, -1, false, false, -1, false, null, null, null, -1, null, null).execute().body();
false, false, false, null, -1, -1, false, false, -1, false, null, null, null, -1, null).execute().body();
System.out.println("predictions: " + predictions);

}
Expand Down
4 changes: 1 addition & 3 deletions h2o-core/src/main/java/water/api/ModelMetricsHandler.java
Original file line number Diff line number Diff line change
Expand Up @@ -173,9 +173,7 @@ public static final class ModelMetricsListSchemaV3 extends RequestSchemaV3<Model

@API(help = "Set number of bins to calculate AUUC. Must be -1 or higher than 0. Default is -1 which means 1000 (optional, only for uplift binomial classification).", json=false, direction = API.Direction.INPUT)
public int auuc_nbins;

@API(help = "Set custom thresholds to calculate AUUC (optional, only for uplift binomial classification). ", json=false, direction = API.Direction.INPUT)
public double[] custom_auuc_thresholds;


// Output fields
@API(help = "ModelMetrics", direction = API.Direction.OUTPUT)
Expand Down

0 comments on commit b490cfa

Please sign in to comment.