Skip to content

Latest commit

 

History

History
30 lines (17 loc) · 1.23 KB

README.md

File metadata and controls

30 lines (17 loc) · 1.23 KB
Raw

Unit 4 - Sprint 14 - Neural Networks

Sketch Classification with TensorFlow Keras

This project builds a sketch classification model using TensorFlow Keras and a sample of the Quickdraw dataset. The dataset has been sampled to only 10 classes and 10000 observations per class.

Objective

The main objective of this project is to build a baseline classification model using TensorFlow Keras and then run a few experiments with different optimizers and learning rates to improve the model's accuracy.

Dataset

The Quickdraw dataset is a collection of over 50 million drawings across 345 categories, collected from the game "Quick, Draw!" The dataset used in this project has been sampled to only 10 classes and 10000 observations per class. The classes are: apple, anvil, airplane, banana, The Eiffel Tower, The Mona Lisa, The Great Wall of China, alarm clock, ant and asparagus.

Requirements

  • Python 3.6 or higher
  • TensorFlow 2.0 or higher
  • NumPy
  • Matplotlib
  • Scikit-learn 1.2.1

Acknowledgements

This project uses the Quickdraw dataset, which was created by Google Creative Lab and is licensed under the Creative Commons Attribution 4.0 International License.

Authors