получить предикторные типы данных из H2OModel - PullRequest
Новая
       10

получить предикторные типы данных из H2OModel

0 голосов
/ 20 сентября 2018

Я знаю, что могу получить доступ к предиктору имен H2OModel через слот @parameters, но могу ли я получить доступ к предиктору типов данных ?

Я пытаюсь сгенерировать схему ввода для моего h2OModel, и сейчас мне нужно сделать перекрестную ссылку на training_frame и получить оттуда типы данных.Очевидно, это было бы проблемой, если бы мой training_frame больше не находился в памяти.

Вот мой текущий подход: 

getInputSchema <- function(model){
  require(jsonlite)
  require(h2o)
  training_frame <- h2o.getFrame(model@parameters$training_frame)
  toJSON(
      setNames( h2o.getTypes(training_frame), 
                names(training_frame)        
               )[model@parameters$x], 
      auto_unbox = T 
      )
}

и пример того, как его можно использовать: 

#--- Example dataset ----
library(h2o)
library(data.table)
options('h2o.use.data.table'=TRUE)
library(rpart.plot) # for 'ptitanic' dataset
h2o.init()
data(ptitanic, package='rpart.plot')
survival <- as.h2o(
              setDT( ptitanic)[, `:=`( age   = as.numeric(age),
                                       sibsp = as.integer(sibsp),
                                       parch = as.integer(parch)
                                     ) ] 
            )
#--- Example model -----
fit <- 
  h2o.gbm( x = c('pclass','sex','age','sibsp','parch'),
           y = 'survived',
           training_frame = survival
          )
#--- Example use ----
getInputSchema(fit)
# {"pclass":"enum","sex":"enum","age":"real","sibsp":"int","parch":"int"}

Я ищу решение, которое можно было бы применить к существующей модели, в которой отсутствует набор данных, указанный в training_frame.

...