本文整理汇总了Java中org.nd4j.linalg.factory.Nd4j.rand方法的典型用法代码示例。如果您正苦于以下问题：Java Nd4j.rand方法的具体用法？Java Nd4j.rand怎么用？Java Nd4j.rand使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类org.nd4j.linalg.factory.Nd4j的用法示例。
在下文中一共展示了Nd4j.rand方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Java代码示例。
示例1: testSmallAmountOfData
import org.nd4j.linalg.factory.Nd4j; //导入方法依赖的package包/类
@Test
public void testSmallAmountOfData(){
MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
.updater(Updater.RMSPROP)
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT).iterations(1)
.list()
.layer(0, new org.deeplearning4j.nn.conf.layers.DenseLayer.Builder()
.nIn(nIn).nOut(3)
.activation("tanh").build())
.layer(1, new org.deeplearning4j.nn.conf.layers.OutputLayer.Builder(LossFunctions.LossFunction.MSE)
.nIn(3).nOut(nOut)
.activation("softmax")
.build())
.build();
SparkDl4jMultiLayer sparkNet = new SparkDl4jMultiLayer(sc,conf,new ParameterAveragingTrainingMaster(true,numExecutors(),1,10,1,0));
Nd4j.getRandom().setSeed(12345);
DataSet d1 = new DataSet(Nd4j.rand(1,nIn),Nd4j.rand(1,nOut));
DataSet d2 = new DataSet(Nd4j.rand(1,nIn),Nd4j.rand(1,nOut));
JavaRDD<DataSet> rddData = sc.parallelize(Arrays.asList(d1,d2));
sparkNet.fit(rddData);
}
开发者ID:PacktPublishing，项目名称:Deep-Learning-with-Hadoop，代码行数:29，代码来源:TestSparkMultiLayerParameterAveraging.java