File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change 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# 动态图训练
5658python 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# 动态图预测
6062python -u ../../../tools/infer.py -m config.yaml -o stage=1 # 全量数据运行config_bigdata.yaml
You can’t perform that action at this time.
0 commit comments