Skip to content

ShenXiexs/rankMixer-encap

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

仓库说明

这个仓库现在保留两个目录,分别对应两种用途。

目录区分

rankmixer-clean

这是整理后的模块化版本。

适合下面这些场景:

  • 想直接把 RankMixer 当成一个可复用模块接到别的项目里
  • 只关心 tokenizer、backbone 和单入口调用
  • 不需要完整训练工程,只需要网络结构本身

这个目录里保留的是:

  • module.py:核心实现
  • __init__.py:包入口
  • README.md:模块使用说明

建议优先使用这个目录。

rankMixer-yoyo-completed

这是之前完整的工程版。

适合下面这些场景:

  • 需要保留原来的训练、评估、导出和推理脚本
  • 需要完整配置、数据管道和业务侧工程代码
  • 想对照之前工程里的训练逻辑和接口

这个目录里保留的是完整工程结构,包括:

  • bin/
  • common/
  • config/
  • dataset/
  • models/
  • rankmixer/

如果只是做模块复用,一般不需要直接改这个目录。

现在推荐怎么用

如果目标是迁移 RankMixer 到其他项目:

  1. 直接使用 rankmixer-clean
  2. rankmixer-clean/README.md 里的方式调用
  3. 需要业务头时,在 pooled_output 后面自己接 CTR、CVR 或其他任务层

如果目标是回看或保留完整工程:

  1. 使用 rankMixer-yoyo-completed
  2. 保留原有训练方式和配置结构

提交约定

根目录 .gitignore 继续生效,不提交这些本地文件:

  • .vscode/
  • __pycache__/
  • *.pyc
  • *.log
  • 参考图.jpg
  • 虚拟环境、临时文件、导出产物

备注

  • 当前仓库主线以 rankmixer-clean 为主
  • rankMixer-yoyo-completed 作为完整工程备份保留

About

RankMixer封装,已去除敏感tokens和features信息

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors