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

fix bug

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