/RuleE

Primary LanguagePython

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