Skip to content

Latest commit

 

History

History
14 lines (14 loc) · 401 Bytes

readme.md

File metadata and controls

14 lines (14 loc) · 401 Bytes
  • main.py
    • RuleE模型入口:训练时执行python main.py
  • train.py
    • 定义loss, optimizer等,并接入模型和dataloader进行训练
  • model.py
    • RuleE模型
  • dataset.py
    • 定义datadet类
  • parameters
    • 其中各个参数文件在训练过程中会生成
  • data
    • 数据文件,data_preprocess模块生成
  • findrules
    • 模型训练完成后用于生成规则