Machine Learning Project (CS420)
Switch branches/tags
Nothing to show
Clone or download
Pull request Compare This branch is even with felixwzh:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
CNN
CapsNet
data
deep_forest
domain_adaptation
report
README.md

README.md

Digit Classification on MNIST

Zhenghui Wang & Ruijie Wang

CNN

  • for training and single model testing,
python mnist_cnn_train.py --version model_version
  • for model ensemble,
python mnist_cnn_test.py --model-dir model_dir --batch-size batch_size --use-ensemble True

CapsNet

  • for training and test:
python mnist_cnn_train.py
  • for image reconstruction:
python capsulenet.py -t -w model/trained_model.h5 --digit digit

Deep Forest

  • for training and test a cascaded deep forest (CA)
python mnist.py --model ca
  • for training and test a cascaded deep forest with mutli-grained scanning (GC)
python mnist.py --model gc

Domain Adaptation

  • for training the source only model
python source_only.py
  • for training MMD
python mmd.py
  • for training DANN
python dann.py
  • for training CORAL
python coral.py
  • for training WDGRL
python wd.py

Reference