Table Of Contents
Table Of Contents

mx.model.buckets

Description

Train RNN with bucket support

Usage

mx.model.buckets(symbol, train.data, eval.data = NULL, metric = NULL,

  arg.params = NULL, aux.params = NULL, fixed.params = NULL,

  num.round = 1, begin.round = 1, initializer = mx.init.uniform(0.01),

  optimizer = "sgd", ctx = NULL, batch.end.callback = NULL,

  epoch.end.callback = NULL, kvstore = "local", verbose = TRUE,

  metric_cpu = TRUE)

Arguments

Argument

Description

symbol

Symbol or list of Symbols representing the model

train.data

Training data created by mx.io.bucket.iter

eval.data

Evaluation data created by mx.io.bucket.iter

num.round

int, number of epoch

verbose