argparse
pytorch
tqdm
matplotlib
numpy (should be installed by pytorch)
scikit-learn
- Install python environment with necessary dependencies
- Activate python enviroment
- Use command line to train model e.g.
python main.py --arg1 4 --arg2 .2 -e 20 -l .0001 -i ./Inputs -o ./Outputs
--arg1 : number of model layers
--arg2 : dropout rate (value from 0.-1.
--epochs (-e) : number of epochs of training
--lr (-l) : learning rate
--output (-o) : output file path
--input (-i) : input file path