嵌入多个字段的deeplearning4j

Question

如何在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中只嵌入了一个字段。如何一次嵌入多个？

Answer 1

在这种情况下，您将使用基于ComputationGraph的网络。一个ComputationGraph可以有多个输入，这将允许您在开始时使用多个嵌入层。

有关这方面的更多信息，请查看文档：https://deeplearning4j.konduit.ai/models/computationgraph#example-2-multiple-inputs-and-merge-vertex