如何使用 scala 将列类型数字更改为苏打水中的枚举

Question

我必须使用 Scala 将数字列更改为苏打水中 h2o 帧的枚举类型以及如何打印 h2o 帧的模式。

Answer 1

您可以使用

.toCategoricalVec

下面是一个用法示例：

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

Answer 2

我们必须转换我们的 H2OFrame。

下面是一个示例：

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