diff --git a/04.word2vec/README.cn.md b/04.word2vec/README.cn.md
index 00493fdf163bfbb8610b4378e72484ed3e3b9a71..5f5d564f9b0ff7abf38a4f9bb9b8a774bdd0f81c 100644
--- a/04.word2vec/README.cn.md
+++ b/04.word2vec/README.cn.md
@@ -337,12 +337,7 @@ def train(use_cuda, train_program, params_dirname):
- `trainer.train`将会开始训练。从`event_handler`返回的监控情况如下:
-```python
-Step 0: Average Cost 7.337213
-Step 10: Average Cost 6.136128
-Step 20: Average Cost 5.766995
-...
-```
+
## 模型应用
在模型训练后,我们可以用它做一些预测。
@@ -394,12 +389,7 @@ def infer(use_cuda, inference_program, params_dirname=None):
在经历3分钟的短暂训练后,我们得到如下的预测。我们的模型预测 `among a group of` 的下一个词是`a`。这比较符合文法规律。如果我们训练时间更长,比如几个小时,那么我们会得到的下一个预测是 `workers`。
-
-```python
-[[0.00106646 0.0007907 0.00072041 ... 0.00049024 0.00041355 0.00084464]]
-6
-a
-```
+
整个程序的入口很简单:
diff --git a/04.word2vec/image/inference_rlt.png b/04.word2vec/image/inference_rlt.png
new file mode 100644
index 0000000000000000000000000000000000000000..4ca8e22d487644e9d62b1e5bacee8b957a0e515c
Binary files /dev/null and b/04.word2vec/image/inference_rlt.png differ
diff --git a/04.word2vec/image/train_rlt.png b/04.word2vec/image/train_rlt.png
new file mode 100644
index 0000000000000000000000000000000000000000..8d0f53f3c04c2b1694fc255e56461bfa2ca21d8a
Binary files /dev/null and b/04.word2vec/image/train_rlt.png differ
diff --git a/04.word2vec/index.cn.html b/04.word2vec/index.cn.html
index e146f6a0f47292134930f4577a57b0fbd9c9ab5b..4b8ca1c6ac78b1bc7621d372c1c6584a4f77788a 100644
--- a/04.word2vec/index.cn.html
+++ b/04.word2vec/index.cn.html
@@ -379,12 +379,7 @@ def train(use_cuda, train_program, params_dirname):
- `trainer.train`将会开始训练。从`event_handler`返回的监控情况如下:
-```python
-Step 0: Average Cost 7.337213
-Step 10: Average Cost 6.136128
-Step 20: Average Cost 5.766995
-...
-```
+
## 模型应用
在模型训练后,我们可以用它做一些预测。
@@ -436,12 +431,7 @@ def infer(use_cuda, inference_program, params_dirname=None):
在经历3分钟的短暂训练后,我们得到如下的预测。我们的模型预测 `among a group of` 的下一个词是`a`。这比较符合文法规律。如果我们训练时间更长,比如几个小时,那么我们会得到的下一个预测是 `workers`。
-
-```python
-[[0.00106646 0.0007907 0.00072041 ... 0.00049024 0.00041355 0.00084464]]
-6
-a
-```
+
整个程序的入口很简单: