我正在尝试Classifier
在 Scala 中编写一个简单的示例,并查看创建实例(使用构建器)然后作为参数传递给类构造函数的dl4j
示例。但是,查看源代码,并没有任何构造函数以 a为例。看起来这些示例不是最新的。有谁知道如何在最后一个版本中做到这一点?NeuralNetConfiguration
MultiLayerNetwork
MultiLayerNetwork
NeuralNetConfiguration
dl4j
我已经设法找到解决方法,但我不确定这是正确的方法:
val conf = new NeuralNetConfiguration.Builder()
.seed(seed)
.iterations(iterations)
.learningRate(1e-6f)
.optimizationAlgo(OptimizationAlgorithm.CONJUGATE_GRADIENT)
.l1(1e-1).regularization(true).l2(2e-4)
.useDropConnect(true)
.layer(hiddenLayer)
.layer(outputLayer)
.build()
val confs: util.List[NeuralNetConfiguration] = new util.LinkedList();
confs.add(conf)
val builder = new MultiLayerConfiguration.Builder()
builder.setConfs(confs)
val model: MultiLayerNetwork = new MultiLayerNetwork(builder.build())