README.md 1.6 KB
Newer Older
Martin Wicke's avatar
Martin Wicke committed
1
# TensorFlow Models
2
3
4
5
6
7

This repository contains machine learning models implemented in
[TensorFlow](https://tensorflow.org). The models are maintained by their
respective authors.

To propose a model for inclusion please submit a pull request.
8
9
10


## Models
11
- [autoencoder](autoencoder) -- various autoencoders
12
13
14
15
16
17
- [inception](inception) -- deep convolutional networks for computer vision
- [namignizer](namignizer) -- recognize and generate names
- [neural_gpu](neural_gpu) -- highly parallel neural computer
- [privacy](privacy) -- privacy-preserving student models from multiple teachers
- [resnet](resnet) -- deep and wide residual networks
- [slim](slim) -- image classification models in TF-Slim
18
- [swivel](swivel) -- the Swivel algorithm for generating word embeddings
19
20
21
- [syntaxnet](syntaxnet) -- neural models of natural language syntax
- [textsum](textsum) -- sequence-to-sequence with attention model for text summarization.
- [transformer](transformer) -- spatial transformer network, which allows the spatial manipulation of data within the network
22
- [im2txt](im2txt) -- image-to-text neural network for image captioning.
Quoc Le's avatar
Quoc Le committed
23

Quoc Le's avatar
Quoc Le committed
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
Implementation of the Neural Programmer model described in https://openreview.net/pdf?id=ry2YOrcge

Download the data from http://www-nlp.stanford.edu/software/sempre/wikitable/
Change the data_dir FLAG to the location of the data

Training:
python neural_programmer.py

The models are written to FLAGS.output_dir 


Testing:
python neural_programmer.py --evaluator_job=True

The models are loaded from FLAGS.output_dir.
The evaluation is done on development data.

Maintained by Arvind Neelakantan (arvind2505)