Skip to content

bannima/ltr_sota

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Learning To Rank SOTA

搜索排序解决方案汇总

Architecture

├── data
│   ├── adult
│   └── census-income
├── doc
├── examples
│   ├── __init__.py
│   ├── adult_deepcross
│   │   ├── README.md
│   │   ├── __init__.py
│   │   ├── adult_deepcross_exp.py
│   ├── adult_widedeep
│   │   ├── README.md
│   │   ├── __init__.py
│   │   ├── adult_widedeep_exp.py
│   └── census_income_mmoe
│       ├── README.md
│       ├── __init__.py
│       ├── census_income_mmoe_exp.py
├── logs
├── ltr
│   ├── __init__.py
│   ├── config.py
│   ├── data_loaders
│   │   ├── __init__.py
|   │   ├── adult_deepcross_loader.py
│   │   ├── adult_loader.py
│   │   ├── basd_loader.py
│   │   └── census_income_loader.py
│   ├── datasets.py
│   ├── features.py
│   ├── models
│   │   ├── __init__.py
│   │   ├── deep_cross.py
│   │   ├── din.py
│   │   ├── layers
│   │   │   ├── __init__.py
│   │   │   ├── core.py
│   │   │   ├── interaction.py
│   │   │   └── sequence.py
│   │   ├── mmoe.py
│   │   └── wide_deep.py
│   ├── modules
│   │   ├── __init__.py
│   │   ├── analyzer.py
│   │   ├── criterions.py
│   │   ├── data_utils.py
│   │   ├── evaluation_metrics.py
│   │   ├── utils.py
│   │   └── visualizer.py
│   ├── project_files
│   └── trainers
│       ├── __init__.py
│       ├── base_trainer.py
│       ├── ltr_trainer.py
│       └── multi_task_trainer.py
├── requirements.txt
├── test

Examples

1. MMoE on Census Income

MMoe on Census Income dataset

python examples/census_income_mmoe/census_income_mmoe_exp.py
  • Experiment Results Analysis

avatar

2. Wide&Deep on Adult

Wide&Deep on Adult Dataset

python examples/adult_widedeep/adult_widedeep_exp.py
  • Experiment Results Analysis

avatar

3. Deep&Cross on Adult

Deep&Cross on Adult Dataset

>>> python adult_deepcross_exp.py
  • Experiment Results Analysis

avatar

4. FM on Adult

Factorization Machines on Adult Dataset

>>> python adult_fm_exp.py
  • Experiment Results Analysis

avatar

LTR Models

Datasets

Reference

About

learning to rank sota solutions

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages