GAT模型用于选品
- 数据预处理模块
- 自定义模型process模块
- 增加batch size模块,避免图太大导致的内存溢出
- 增加focal loss模块用于不均衡数据
数据预处理
python3 preprocess.py --augment模型训练
python3 run.py --save --cuda --batch_size 10000 --augmentpaper link:https://arxiv.org/abs/1710.10903
reference code:https://github.com/dmlc/dgl/tree/master/examples/pytorch/gat
blog:https://zhuanlan.zhihu.com/p/81350196