/model-zoo

Please do not feed the models

Primary LanguageJuliaOtherNOASSERTION

Flux Model Zoo

This repository contains various demonstrations of the Flux machine learning library. Any of these may freely be used as a starting point for your own models.

  • housing implements the most basic model possible (a linear regression) on the UCI housing data set. It's bare-bones and illustrates how to build a model from scratch.
  • mnist classifies digits from the MNIST data set, using a simple multi-layer perceptron and a convolutional network, as well as showing a simple autoencoder.
  • char-rnn implements a character-level language model. It comes with a Shakespeare dataset but can work with any text.
  • phonemes implements a sequence to sequence model with attention, using the CMU pronouncing dictionary to predict the pronunciations of unknown words.
  • lang-detection implements a simple sequence-to-classification model, which recognises language (English, Danish etc.) from input characters.
  • treebank shows a recursive neural network with the Stanford Sentiment Treebank

Note that these models are best run line-by-line, either in the REPL or Juno.