- using pip
git clone https://github.com/pydsgz/base-ml.git
cd base-ml
pip install -r requirements.txt
Dataset should be in
./data/<dataset_name>/
python train.py --dataset=<dataset_name> --exp_num=1
For example, to run experiment using dizzyreg-task 1 dataset. exp_num is a
unique number added to output folders.
python train.py --dataset=dizzyreg1 --exp_num=1
- All outputs from this experiment will be in
./examples/outputs/dizzyreg1/exp_001/
You can plot classifiation results using the notebooks in ./notebooks/
cd ./notebooks/
jupyter notebook