Skip to main content

`prior`

  • Available in: GLM, GAM
  • Hyperparameter: no

Description​

This option specifies the prior probability of class 1 in the response when family = "binomial". This must be a value from the (0,1) exclusive range. The API default is -1, which means no prior is applied and the observation frequency of class 1 is used. This parameter is useful for logistic regression if the data has been sampled and the mean of response does not reflect reality.

  • None

Example​

library(h2o)
h2o.init()

# import the cars dataset:
# this dataset is used to classify whether or not a car is economical based on
# the car's displacement, power, weight, and acceleration, and the year it was made
cars <- h2o.importFile("https://s3.amazonaws.com/h2o-public-test-data/smalldata/junit/cars_20mpg.csv")

# convert response column to a factor
cars["economy_20mpg"] <- as.factor(cars["economy_20mpg"])

# set the predictor names and the response column name
predictors <- c("displacement", "power", "weight", "acceleration", "year")
response <- "economy_20mpg"

# split into train and validation
cars_splits <- h2o.splitFrame(data = cars, ratios = 0.8)
train <- cars_splits[[1]]
valid <- cars_splits[[2]]

# Build a GLM model and set a prior value of 0.5
car_glm1 <- h2o.glm(x = predictors, y = response, family = 'binomial', prior=0.5,
                    training_frame = train, 
                    validation_frame = valid)

# Build a GLM model without a prior value
car_glm2 <- h2o.glm(x = predictors, y = response, family = 'binomial',
                    training_frame = train,
                    validation_frame = valid)

# Check the coefficients for both models
car_glm1@model$coefficients_table
car_glm2@model$coefficients_table
Feedback