All Downloads are FREE. Search and download functionalities are using the official Maven repository.

hex.schemas.NaiveBayesV3 Maven / Gradle / Ivy

package hex.schemas;

import hex.naivebayes.NaiveBayes;
import hex.naivebayes.NaiveBayesModel;
import water.api.API;
import water.api.SupervisedModelParametersSchema;

public class NaiveBayesV3 extends SupervisedModelBuilderSchema {
  public static final class NaiveBayesParametersV3 extends SupervisedModelParametersSchema {
    static public String[] own_fields = new String[]{"laplace", "min_sdev", "eps_sdev", "min_prob", "eps_prob", "compute_metrics"};

    @API(help = "Laplace smoothing parameter")
    public double laplace;

    @API(help = "Min. standard deviation to use for observations with not enough data")
    public double min_sdev;

    @API(help = "Cutoff below which standard deviation is replaced with min_sdev")
    public double eps_sdev;

    @API(help = "Min. probability to use for observations with not enough data")
    public double min_prob;

    @API(help = "Cutoff below which probability is replaced with min_prob")
    public double eps_prob;

    @API(help = "Compute metrics on training data")
    public boolean compute_metrics;
  }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy