/stl_10_classifier

Primary LanguageJupyter Notebook

stl_10_classifier

final_trainer.ipynb

  • Used for competition.
  • Modify the "ROOT_DIR" variable to point to the "cse7512-00-dl-tp-pt1.zip".

data_helper.ipynb

  • Used for dividing the original training data into training and validation data.
  • Modify the "ROOT_DIR" variable to point to the "cse7512-00-dl-tp-pt1.zip".
  • It will output "processed_data.zip".
  • "processed_data.zip" contains training, validation and testing folders.

baseline.ipynb

  • Used for testing different models.
  • It uses the "processed_data.zip".
  • Takes advantage of validation data to check how the model is performing.
  • Modify the "ROOT_DIR" variable to point to the "processed_data.zip".