首页 文章

H2O深度学习R [关闭]

提问于
浏览
1

H2O Deep Learning默认运行回归,即使我确保目标变量是一个因子(只有两个级别) . 有关如何解决此问题的任何线索?

以下是代码:

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)

1 回答

  • 2

    如果要运行分类,则必须将响应变量编码为"factor"(aka "enum")类型 . 请参阅H2O Deep Learning booklet中的R code example . 所有H2O算法都是如此 .

相关问题