Skip to content

Commit 3cd246d

Browse files
committed
fix autofis readme
1 parent 64470fa commit 3cd246d

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

models/rank/autofis/readme.md

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,19 @@
55
```
66
├── data #样例数据
77
├── sample_data #样例数据
8-
├── train
9-
├── train_x.npy #训练数据样例
10-
├── train_y.npy #训练数据样例
8+
├── sample_train_x.txt #训练数据样例
9+
├── sample_train_y.txt #训练数据样例
1110
├── __init__.py
12-
├── README.md #文档
1311
├── config.yaml # sample数据配置
1412
├── config_bigdata.yaml # 全量数据配置
15-
├── net.py # 模型核心组网
1613
├── criteo_reader.py #数据读取程序
1714
├── dygraph_model.py # 构建动态图
15+
├── metrics.py # 自定义评估指标
16+
├── net.py # 模型核心组网
17+
├── optimizer.py # 自定义优化器
1818
├── trainer.py # 训练脚本
19+
├── README.md #文档
20+
├── trainer.py #训练入口
1921
```
2022

2123
注:在阅读该示例前,建议您先了解以下内容:
@@ -54,7 +56,7 @@ os : windows/linux/macos
5456
# cd models/rank/autofis # 在任意目录均可运行
5557
# 动态图训练
5658
python trainer.py -m config.yaml # stage0:自动搜索最佳特征组合 全量数据运行config_bigdata.yaml
57-
python trainer.py -m config.yaml -o stage=1 # stage1:训练最终模 全量数据运行config_bigdata.yaml型
59+
python trainer.py -m config.yaml -o stage=1 # stage1:训练最终模型 全量数据运行config_bigdata.yaml
5860

5961
# 动态图预测
6062
python -u ../../../tools/infer.py -m config.yaml -o stage=1 # 全量数据运行config_bigdata.yaml

0 commit comments

Comments
 (0)