H2O Deep Learning is running regression by default even though I have ensured that the target variable is a factor (with only two levels). Any leads on how to resolve this ?
Below is the code :
dnn_mod <-
h2o.deeplearning(x = 2:321, # column numbers for predictors
y = 322, # column number for label
training_frame = sdcs_data, # data in H2O format
activation = "TanhWithDropout", # or 'Tanh'
input_dropout_ratio = 0.2, # % of inputs dropout
hidden_dropout_ratios = c(0.3,0.3,0.3), # % for nodes dropout
balance_classes = FALSE,
hidden = c(150,150,150),
epochs = 500,
#standardize = TRUE,
epsilon = 1.0e-5,
loss = "CrossEntropy",
stopping_rounds = 50,
stopping_metric = "AUC")
#classification = TRUE)