execute the following command from the project home directory:
python NJM.py --dataset 1 --dimensions 10 --epochs 20
You can check out the other options available to use with NJM using:
python NJM.py --help
- Python version: '3.6.3'
- TensorFlow version: '1.0.1'
- Numpy version: '1.14.0'