文件名称:ranknet-tensorflow2.0:TensorFlow 2.0中RankNet到LambdaRank的实现
文件大小:13KB
文件格式:ZIP
更新时间:2024-06-11 23:27:11
tensorflow ltr learning-to-rank ranknet lambdarank
TensorFlow2.0中Ranknet到LambdaRank的实现 **此存储库在Tensorflow 2.0中具有从RankNet到LambdaRank的实现,** 要求 tqdm == 4.32.1 numpy == 1.16.4 点击== 7.0 tensorflow_gpu == 2.1.0 设置 $ git clone https://github.com/akanyaani/ranknet-tensorflow2.0 $ cd ranknet-tensorflow2.0 $ pip install -r requirements.txt 从此处下载数据,并将任何折叠传递给pre_process。 $ python pre_process.py --help Options: --data-dir TEXT training data
【文件预览】:
ranknet-tensorflow2.0-master
----.gitignore(1KB)
----data_prep.ipynb(8KB)
----train_model.py(2KB)
----requirements.txt(56B)
----ranknet.py(11KB)
----utils.py(192B)
----pre_process.py(2KB)
----README.md(3KB)
----metrics.py(920B)
----layer_norm.py(1KB)
----data_pipeline.py(2KB)
----lambdarank.py(2KB)