Skip to content

Jupyter notebooks for learning how to develop and deploy ML models

Notifications You must be signed in to change notification settings

TreeHacks/hackpack-ml

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 

Repository files navigation

hackpack-ml

Learn how to make and deploy ML hacks in PyTorch.

Features

This hackpack aims to make iterating on your models easy while also providing for quick integration into apps.

We include:

  1. An introductory notebook in which we load our own dataset into PyTorch and build a simple neural network for regression on tabular data: /models/dataset_nn/dataset_neural_nets.ipynb

  2. An application-focused notebook in which we load and train a DenseNet model for image classification on MNIST and CIFAR10 datasets, and then export the model into a Flask server as our own API: /models/densenet_web/densenet.ipynb

Getting Started

This hackpack assumes some prior understanding of Python. Understanding of Deep Learning techniques is greatly beneficial but not required. If you are interested in learning, Deep Learning by Goodfellow et al. is a good starting point.

Google Colab

Both notebooks are also hosted on Google Colab, a free Jupyter notebook environment with GPU/TPU support. They are accessible at dataset_neural_nets and densenet_web respectively. Feel free to open them in the playground environment or copy!

Installing

If you wish to work on the notebooks locally, the dependencies are listed in environment.yml. Create a new Conda environment with:

conda env create --file=environment.yml

License

MIT

About HackPacks 🌲

HackPacks are built by the TreeHacks team to help hackers build great projects at our hackathon that happens every February at Stanford. We believe that everyone of every skill level can learn to make awesome things, and this is one way we help facilitate hacker culture. We open source our hackpacks (along with our internal tech) so everyone can learn from and use them! Feel free to use these at your own hackathons, workshops, and anything else that promotes building :)

If you're interested in attending TreeHacks, you can apply on our website during the application period.

You can follow us here on GitHub to see all the open source work we do (we love issues, contributions, and feedback of any kind!), and on Facebook, Twitter, and Instagram to see general updates from TreeHacks.

About

Jupyter notebooks for learning how to develop and deploy ML models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published