Commit 3bbc5d2f authored by Neal Wu's avatar Neal Wu
Browse files

Make the README for textsum a little clearer

parent 29881fb4
...@@ -70,9 +70,7 @@ vocabulary size: Most frequent 200k words from dataset's article and summaries. ...@@ -70,9 +70,7 @@ vocabulary size: Most frequent 200k words from dataset's article and summaries.
<b>How To Run</b> <b>How To Run</b>
Pre-requesite: Prerequisite: install TensorFlow and Bazel.
Install TensorFlow and Bazel.
```shell ```shell
# cd to your workspace # cd to your workspace
...@@ -83,7 +81,7 @@ Install TensorFlow and Bazel. ...@@ -83,7 +81,7 @@ Install TensorFlow and Bazel.
# If your data files have different names, update the --data_path. # If your data files have different names, update the --data_path.
# If you don't have data but want to try out the model, copy the toy # If you don't have data but want to try out the model, copy the toy
# data from the textsum/data/data to the data/ directory in the workspace. # data from the textsum/data/data to the data/ directory in the workspace.
ls -R $ ls -R
.: .:
data textsum WORKSPACE data textsum WORKSPACE
...@@ -97,38 +95,38 @@ data.py seq2seq_attention_decode.py seq2seq_attention.py seq2seq_lib.py ...@@ -97,38 +95,38 @@ data.py seq2seq_attention_decode.py seq2seq_attention.py seq2seq_lib.py
./textsum/data: ./textsum/data:
data vocab data vocab
bazel build -c opt --config=cuda textsum/... $ bazel build -c opt --config=cuda textsum/...
# Run the training. # Run the training.
bazel-bin/textsum/seq2seq_attention \ $ bazel-bin/textsum/seq2seq_attention \
--mode=train \ --mode=train \
--article_key=article \ --article_key=article \
--abstract_key=abstract \ --abstract_key=abstract \
--data_path=data/training-* \ --data_path=data/training-* \
--vocab_path=data/vocab \ --vocab_path=data/vocab \
--log_root=textsum/log_root \ --log_root=textsum/log_root \
--train_dir=textsum/log_root/train --train_dir=textsum/log_root/train
# Run the eval. Try to avoid running on the same machine as training. # Run the eval. Try to avoid running on the same machine as training.
bazel-bin/textsum/seq2seq_attention \ $ bazel-bin/textsum/seq2seq_attention \
--mode=eval \ --mode=eval \
--article_key=article \ --article_key=article \
--abstract_key=abstract \ --abstract_key=abstract \
--data_path=data/validation-* \ --data_path=data/validation-* \
--vocab_path=data/vocab \ --vocab_path=data/vocab \
--log_root=textsum/log_root \ --log_root=textsum/log_root \
--eval_dir=textsum/log_root/eval --eval_dir=textsum/log_root/eval
# Run the decode. Run it when the most is mostly converged. # Run the decode. Run it when the most is mostly converged.
bazel-bin/textsum/seq2seq_attention \ $ bazel-bin/textsum/seq2seq_attention \
--mode=decode \ --mode=decode \
--article_key=article \ --article_key=article \
--abstract_key=abstract \ --abstract_key=abstract \
--data_path=data/test-* \ --data_path=data/test-* \
--vocab_path=data/vocab \ --vocab_path=data/vocab \
--log_root=textsum/log_root \ --log_root=textsum/log_root \
--decode_dir=textsum/log_root/decode \ --decode_dir=textsum/log_root/decode \
--beam_size=8 --beam_size=8
``` ```
...@@ -157,7 +155,7 @@ article: the european court of justice ( ecj ) recently ruled in lock v british ...@@ -157,7 +155,7 @@ article: the european court of justice ( ecj ) recently ruled in lock v british
abstract: will british gas ecj ruling fuel holiday pay hike ? abstract: will british gas ecj ruling fuel holiday pay hike ?
decode: eu law requires worker 's statutory holiday pay decode: eu law requires worker 's statutory holiday pay
====================================== ======================================
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment