如何在deeplearning4j中嵌入多个字段?
例如,如果我想嵌入 user_id
以及 ad_id
作为两个字段:
val conf = new NeuralNetConfiguration.Builder()
.seed(12345)
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT).iterations(1)
.activation(Activation.LEAKYRELU)
.weightInit(WeightInit.XAVIER)
.learningRate(0.01)
.updater(Updater.NESTEROVS).momentum(0.9)
.regularization(true).l2(1e-4)
.list
.layer(0, new EmbeddingLayer.Builder().nIn(FEATURE_DIMS).nOut(512).activation(Activation.IDENTITY).build())
.layer(1, new DenseLayer.Builder().nIn(512).nOut(10).build)
.layer(2, new OutputLayer.Builder(LossFunctions.LossFunction.NEGATIVELOGLIKELIHOOD)
.activation(Activation.SOFTMAX)
.nIn(10).nOut(1).build)
.pretrain(false)
.backprop(true)
.build
此代码只在deeplearning4j中嵌入一个字段。如何一次嵌入多个?
1条答案
按热度按时间inn6fuwd1#
在这种情况下,您将使用基于图形的网络。computationgraph可以有多个输入,这将允许您在开始时使用多个嵌入层。
有关这方面的更多信息,请参阅文档:https://deeplearning4j.konduit.ai/models/computationgraph#example-2多输入合并顶点