如果你想尝试训练MLP,可以在codes文件夹下终端输入:
python train_model.py --model MLP --lr ? --gamma ?
其中 , ? 指你想要赋予的值。
如果你想尝试训练CNN,可以在codes文件夹下终端输入:
python train_model.py --model CNN --lr ? --gamma ?
如果你想尝试测试MLP,可以在codes文件夹下终端输入:
python test_model.py --path MLP
如果你想尝试测试CNN,可以在codes文件夹下终端输入:
python test_model.py --path CNN
如果你想训练不同hidden layer的MLP或多个卷积核的CNN,请自己尝试修改代码。