diff --git a/h2o-admissibleml/src/main/java/hex/Infogram/InfogramModel.java b/h2o-admissibleml/src/main/java/hex/Infogram/InfogramModel.java index 9d8a09c127ec..bd723b14a186 100644 --- a/h2o-admissibleml/src/main/java/hex/Infogram/InfogramModel.java +++ b/h2o-admissibleml/src/main/java/hex/Infogram/InfogramModel.java @@ -5,11 +5,10 @@ import hex.*; import hex.genmodel.utils.DistributionFamily; import hex.glm.GLMModel; -import hex.schemas.*; +import hex.schemas.InfogramV3; import water.*; import water.fvec.Frame; import water.udf.CFuncRef; -import water.util.TwoDimTable; import java.lang.reflect.Field; import java.util.*; @@ -55,10 +54,10 @@ public static class InfogramParameters extends Model.Parameters { public String[] _protected_columns = null; // store features to be excluded from final model public double _cmi_threshold = 0.1; // default set by Deep public double _relevance_threshold = 0.1; // default set by Deep - public double _total_information_threshold = -1; // relevance threshold for core infogram - public double _net_information_threshold = -1; // cmi threshold for core infogram - public double _safety_index_threshold = -1; // cmi threshold for safe infogram - public double _relevance_index_threshold = -1; // relevance threshold for safe infogram + public double _total_information_threshold = 0.1; // relevance threshold for core infogram + public double _net_information_threshold = 0.1; // cmi threshold for core infogram + public double _safety_index_threshold = 0.1; // cmi threshold for safe infogram + public double _relevance_index_threshold = 0.1; // relevance threshold for safe infogram public double _data_fraction = 1.0; // fraction of data to use to calculate infogram public Model.Parameters _infogram_algorithm_parameters; // store parameters of chosen algorithm public int _top_n_features = 50; // if 0 consider all predictors, otherwise, consider topk predictors diff --git a/h2o-admissibleml/src/main/java/hex/schemas/InfogramV3.java b/h2o-admissibleml/src/main/java/hex/schemas/InfogramV3.java index 1ca6c2dff9c0..55ac12d2032a 100644 --- a/h2o-admissibleml/src/main/java/hex/schemas/InfogramV3.java +++ b/h2o-admissibleml/src/main/java/hex/schemas/InfogramV3.java @@ -15,12 +15,10 @@ import water.api.SchemaServer; import water.api.schemas3.KeyV3; import water.api.schemas3.ModelParametersSchemaV3; -import static hex.util.DistributionUtils.distributionToFamily; + import java.util.*; -import java.util.Arrays; -import java.util.HashMap; -import java.util.Map; -import java.util.Properties; + +import static hex.util.DistributionUtils.distributionToFamily; public class InfogramV3 extends ModelBuilderSchema { public static final class InfogramParametersV3 extends ModelParametersSchemaV3 { @@ -134,44 +132,43 @@ public static final class InfogramParametersV3 extends ModelParametersSchemaV3