Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
5a9951a8
Unverified
Commit
5a9951a8
authored
Feb 26, 2018
by
Andrew M Dai
Committed by
GitHub
Feb 26, 2018
Browse files
Merge pull request #3447 from a-dai/master
Fix extra -- when not using the bazel command line.
parents
a2f1732d
bc16e139
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
11 deletions
+12
-11
research/adversarial_text/README.md
research/adversarial_text/README.md
+12
-11
No files found.
research/adversarial_text/README.md
View file @
5a9951a8
...
...
@@ -10,7 +10,7 @@ Code for [*Adversarial Training Methods for Semi-Supervised Text Classification*
### Fetch data
```
```
bash
$
wget http://ai.stanford.edu/~amaas/data/sentiment/aclImdb_v1.tar.gz
\
-O
/tmp/imdb.tar.gz
$
tar
-xf
/tmp/imdb.tar.gz
-C
/tmp
...
...
@@ -20,9 +20,9 @@ The directory `/tmp/aclImdb` contains the raw IMDB data.
### Generate vocabulary
```
```
bash
$ IMDB_DATA_DIR
=
/tmp/imdb
$ python gen_vocab.py
--
\
$
python gen_vocab.py
\
--output_dir
=
$IMDB_DATA_DIR
\
--dataset
=
imdb
\
--imdb_input_dir
=
/tmp/aclImdb
\
...
...
@@ -33,8 +33,8 @@ Vocabulary and frequency files will be generated in `$IMDB_DATA_DIR`.
### Generate training, validation, and test data
```
$ python gen_data.py
--
\
```
bash
$
python gen_data.py
\
--output_dir
=
$IMDB_DATA_DIR
\
--dataset
=
imdb
\
--imdb_input_dir
=
/tmp/aclImdb
\
...
...
@@ -46,9 +46,9 @@ $ python gen_data.py -- \
### Pretrain IMDB Language Model
```
```
bash
$ PRETRAIN_DIR
=
/tmp/models/imdb_pretrain
$ python pretrain.py
--
\
$
python pretrain.py
\
--train_dir
=
$PRETRAIN_DIR
\
--data_dir
=
$IMDB_DATA_DIR
\
--vocab_size
=
86934
\
...
...
@@ -74,9 +74,9 @@ addition of `pretrained_model_dir`, from which the classifier will load the
pretrained embedding and LSTM variables, and flags related to adversarial
training and classification.
```
```
bash
$ TRAIN_DIR
=
/tmp/models/imdb_classify
$ python train_classifier.py
--
\
$
python train_classifier.py
\
--train_dir
=
$TRAIN_DIR
\
--pretrained_model_dir
=
$PRETRAIN_DIR
\
--data_dir
=
$IMDB_DATA_DIR
\
...
...
@@ -99,9 +99,9 @@ $ python train_classifier.py -- \
### Evaluate on test data
```
```
bash
$ EVAL_DIR
=
/tmp/models/imdb_eval
$ python evaluate.py
--
\
$
python evaluate.py
\
--eval_dir
=
$EVAL_DIR
\
--checkpoint_dir
=
$TRAIN_DIR
\
--eval_data
=
test
\
...
...
@@ -154,3 +154,4 @@ control which dataset is processed and how.
*
Ryan Sepassi, @rsepassi
*
Andrew M. Dai, @a-dai
<adai@google.com>
*
Takeru Miyato, @takerum (Original implementation)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment