/numpy-mlp

Primary LanguagePython

numpy-mlp

Usage:

  1. Install required packages
pip install -r requirements.txt
  1. Run training
python mlp.py [-h] [--data_dir DATA_DIR] [--hidden_dims HIDDEN_DIMS [HIDDEN_DIMS ...]] [--epochs EPOCHS] [--batch_size BATCH_SIZE] [--learning_rate LEARNING_RATE]