This is a general neural network training & testing framework that I use in personal and course projects. It uses Keras on top of TensorFlow. You may use this framework subject to the license.
Step 1: Install Python 3.6/3.7 and the corresponding pip. The framework may also run on Python 3.8+.
Step 2: Install requirements: pip3 install -r resource/requirements.txt
Command to run:
python start_scripts.py [-h] [-t TASK] [-m MODEL_PATH]
Set TASK depending on the specific task to run:
- 0: print model summary
- 1: train
- 2: test
- 3: train + test
Set MODEL_PATH as the model file path to save to or load from.
Use -h flag to see details about these arguments.
12/22/2019: Opened repo. Finished basic neural network framework.