Add functionality to construct and train RNNs

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning

Apache License 2.0

3.18k stars 439 forks source link

Add functionality to construct and train RNNs #5

Closed ecod3r closed 8 years ago

ecod3r commented 8 years ago

As per title, it would be great to have the ability to build and train RNNs easily with this library.

ilblackdragon commented 8 years ago

It's currently possible to train RNNs (see text_classification* examples), but it would be nice to have semi-generic RNNClassifier / RNNRegressor.

terrytangyuan commented 8 years ago

@ecoskian Just added this functionality. See https://github.com/tensorflow/skflow/blob/master/examples/text_classification_builtin_rnn_model.py You can also change the cell_type to what you need, e.g. rnn, gru, and lstm.