Skip to content

Commit f1e596e

Browse files
committed
Merge branch 'master' of https://github.com/tech-srl/code2vec
2 parents 8ae3a83 + e389fdc commit f1e596e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,11 +168,11 @@ In order to export embeddings from a trained model, use the "--save_w2v" and "--
168168

169169
Exporting the trained *token* embeddings:
170170
```
171-
python3 code2vec.py --load models/java14_model/saved_model_iter3 --save_w2v models/java14_model/tokens.txt
171+
python3 code2vec.py --load models/java14_model/saved_model_iter8 --save_w2v models/java14_model/tokens.txt
172172
```
173173
Exporting the trained *target* (method name) embeddings:
174174
```
175-
python3 code2vec.py --load models/java14_model/saved_model_iter3 --save_t2v models/java14_model/targets.txt
175+
python3 code2vec.py --load models/java14_model/saved_model_iter8 --save_t2v models/java14_model/targets.txt
176176
```
177177
This saves the tokens/targets embedding matrices in word2vec format to the specified text file, in which:
178178
the first line is: \<vocab_size\> \<dimension\>

0 commit comments

Comments
 (0)