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 -``` +
整个程序的入口很简单: