提交 f3042c58 编写于 作者: zhaoyijin666's avatar zhaoyijin666

fix bug

上级 30d49cd2
...@@ -90,6 +90,7 @@ optional arguments: ...@@ -90,6 +90,7 @@ optional arguments:
例如可执行下列命令, 完成数据预处理: 例如可执行下列命令, 完成数据预处理:
```shell ```shell
mkdir output
python data_processor.py --train_set_path=./data/train.txt \ python data_processor.py --train_set_path=./data/train.txt \
--output_dir=./output \ --output_dir=./output \
--feat_appear_limit=20 --feat_appear_limit=20
...@@ -255,6 +256,7 @@ reader = Reader(feature_dict, args.window_size) ...@@ -255,6 +256,7 @@ reader = Reader(feature_dict, args.window_size)
``` ```
接下去就可以开始训练了,可执行以下命令: 接下去就可以开始训练了,可执行以下命令:
```shell ```shell
mkdir output/model
python train.py --train_set_path='./data/train.txt' \ python train.py --train_set_path='./data/train.txt' \
--test_set_path='./data/test.txt' \ --test_set_path='./data/test.txt' \
--model_output_dir='./output/model/' \ --model_output_dir='./output/model/' \
......
...@@ -79,6 +79,7 @@ The fucntion of this script is as follows: ...@@ -79,6 +79,7 @@ The fucntion of this script is as follows:
For example, run the following command to accomplish data pre-processing: For example, run the following command to accomplish data pre-processing:
```shell ```shell
mkdir output
python data_processor.py --train_set_path=./data/train.txt \ python data_processor.py --train_set_path=./data/train.txt \
--output_dir=./output \ --output_dir=./output \
--feat_appear_limit=20 --feat_appear_limit=20
...@@ -193,7 +194,6 @@ return paddle.layer.nce( ...@@ -193,7 +194,6 @@ return paddle.layer.nce(
num_classes=len(self._feature_dict['history_clicked_items']), num_classes=len(self._feature_dict['history_clicked_items']),
param_attr=paddle.attr.Param(name="nce_w"), param_attr=paddle.attr.Param(name="nce_w"),
bias_attr=paddle.attr.Param(name="nce_b"), bias_attr=paddle.attr.Param(name="nce_b"),
act=paddle.activation.Sigmoid(),
num_neg_samples=5, num_neg_samples=5,
neg_distribution=self._item_freq) neg_distribution=self._item_freq)
``` ```
...@@ -240,6 +240,7 @@ reader = Reader(feature_dict, args.window_size) ...@@ -240,6 +240,7 @@ reader = Reader(feature_dict, args.window_size)
``` ```
Then start training. Then start training.
```shell ```shell
mkdir output/model
python train.py --train_set_path='./data/train.txt' \ python train.py --train_set_path='./data/train.txt' \
--test_set_path='./data/test.txt' \ --test_set_path='./data/test.txt' \
--model_output_dir='./output/model/' \ --model_output_dir='./output/model/' \
......
...@@ -147,7 +147,6 @@ class DNNmodel(object): ...@@ -147,7 +147,6 @@ class DNNmodel(object):
num_classes=len(self._feature_dict['history_clicked_items']), num_classes=len(self._feature_dict['history_clicked_items']),
param_attr=paddle.attr.Param(name="nce_w"), param_attr=paddle.attr.Param(name="nce_w"),
bias_attr=paddle.attr.Param(name="nce_b"), bias_attr=paddle.attr.Param(name="nce_b"),
act=paddle.activation.Sigmoid(),
num_neg_samples=5, num_neg_samples=5,
neg_distribution=self._item_freq) neg_distribution=self._item_freq)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册