/FashionableLearning

An application of multiple machine learning architectures in TensorFlow on the Fashion-MNIST dataset

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

How to train a model

  1. Change directory to FashionableLearning/main/
  2. $ python main.py -c [model_name]

Example: To train the logistic regression model, type $ python main.py -c logistic_regression

How to visualize a trained model in tensorboard

  1. $ tensorboard --logdir=[path_to_log_directory]
  2. Go to http://localhost:6006/

Project architecture based on https://github.com/MrGemy95/Tensorflow-Project-Template