关于深度学习框架mxnet R包，如何以精度为基准停止迭代

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

最近在使用mxnet学习框架，数据集有很多，每个数据集的最优迭代数不同，每次都需要穷举逼近最优的跌代数，但是观测得到，每次快到最优迭代数的时候，精度总是逼近一个值，现在想如何精度到一个值的时候自动停止，附一段建模部分代码

print("Setting model...")
  data <- mx.symbol.Variable("data")
  fc1 <- mx.symbol.FullyConnected(data, name="fc1", num_hidden=500)
  act1 <- mx.symbol.Activation(fc1, name="relu1", act_type="sigmoid")
  fc2 <- mx.symbol.FullyConnected(act1, name="fc2", num_hidden=256)
  act2 <- mx.symbol.Activation(fc2, name="relu2", act_type="relu")
  fc3 <- mx.symbol.FullyConnected(act2, name="fc3", num_hidden=3)
  #act3 <- mx.symbol.Activation(fc3, name="relu3", act_type="relu")
  #fc4 <- mx.symbol.FullyConnected(act3, name="fc4", num_hidden=10)
  softmax <- mx.symbol.SoftmaxOutput(fc3, name="sm")

  devices <- mx.gpu()


  mx.set.seed(0)
  DNN_model <- mx.model.FeedForward.create(softmax, X=train.x, y=train.y,
                                           ctx=devices, num.round=170, array.batch.size=300,
                                           learning.rate=learning.rate, momentum=0.9, eval.metric=RE_evaluate,
                                           initializer=mx.init.uniform(0.01),
                                           epoch.end.callback=mx.callback.log.train.metric(100))

  preds <- predict(DNN_model, test)

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏3 回帖

关键词：深度学习 NET setting model print 如何