csviter mxnet

最新推荐文章于 2024-09-21 22:38:11 发布

qq_28151575

最新推荐文章于 2024-09-21 22:38:11 发布

阅读量655

点赞数

分类专栏： ml 文章标签： mxnet

本文链接：https://blog.csdn.net/qq_28151575/article/details/51569366

版权

ml 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

import mxnet as mx

def mlp():
    data=mx.sym.Variable('data')
    fc1=mx.sym.FullyConnected(data, name="fc1", num_hidden=512)
    act1=mx.sym.Activation(fc1, name="relu1", act_type="relu")
    fc2=mx.sym.FullyConnected(act1, name="fc2", num_hidden=512)
    act2=mx.sym.Activation(fc2, name="relu2", act_type="relu")
    fc3=mx.sym.FullyConnected(act2, name="fc3", num_hidden=10)
    mlp=mx.sym.SoftmaxOutput(fc3, name="softmax")
    return mlp 

if __name__=="__main__":
    num_epoch=3
    batch_size=100
    train_dataiter=mx.io.CSVIter(data_csv="mnist.train", data_shape=(28, 28), label_csv="label.train", label_shape=(1,), batch_size=batch_size)
    val_dataiter=mx.io.CSVIter(data_csv="mnist.val", data_shape=(28, 28), label_csv="label.val", label_shape=(1,), batch_size=batch_size)
    mlp=mlp()
    mod=mx.mod.Module(mlp)
    #bind and init_params
    mod.bind(data_shapes=train_dataiter.provide_data, label_shapes=train_dataiter.provide_label)
    mod.init_params()

    mod.init_optimizer(optimizer_params={'learning_rate':0.01, 'momentum': 0.9})
    metric = mx.metric.create('acc')

    for i_epoch in range(num_epoch):
        for i_iter, batch in enumerate(train_dataiter):
            mod.forward(batch)
            mod.update_metric(metric, batch.label)

            mod.backward()
            mod.update()

        for name, val in metric.get_name_value():
            print('epoch %03d: %s=%f' % (i_epoch, name, val))
        metric.reset()
        train_dataiter.reset()