Как изменить тип столбца числовой на enum в газированной воде, используя scala?

Question

Мне нужно изменить числовые столбцы на тип Enum для рамки h2o в газированной воде с использованием Scala и способа печати схемы рамки h2o.

Answer 1

Мы должны преобразовать наш H2OFrame.

Вот пример:

NameOfH2OFrame.colToEnum(Array(<"NameOfColumnYouWantToChangeInEnum">))

val h2oFrameTrain = h2oContext.asH2OFrame(trainingData)
h2oFrameTrain.colToEnum(Array("WEEKDAY_LABEL", "EVENT_TYPE", "EVENT_NAME"))

Answer 2

Вы можете использовать .toCategoricalVec

Вот пример использования:

val trainFrame:H2OFrame = bigTable
withLockAndUpdate(trainFrame){ fr => fr.replace(19, fr.vec("IsDepDelayed").toCategoricalVec)}