├── README.md ├── README_zh.md ├── dbrl ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── process.py │ ├── session.py │ └── split.py ├── evaluate │ ├── __init__.py │ ├── evaluate.py │ └── metrics.py ├── models │ ├── __init__.py │ ├── bcq.py │ ├── ddpg.py │ ├── dssm.py │ └── youtube_topk.py ├── network │ ├── __init__.py │ └── net.py ├── resources │ └── aa ├── run_bcq.py ├── run_ddpg.py ├── run_prepare_data.py ├── run_pretrain_embeddings.py ├── run_reinforce.py ├── serialization │ ├── __init__.py │ └── serialize.py ├── trainer │ ├── __init__.py │ ├── pretrain.py │ └── train.py └── utils │ ├── __init__.py │ ├── info.py │ ├── misc.py │ ├── params.py │ └── sampling.py └── pic ├── readme2.png └── readme5.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/README.md -------------------------------------------------------------------------------- /README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/README_zh.md -------------------------------------------------------------------------------- /dbrl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dbrl/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/data/__init__.py -------------------------------------------------------------------------------- /dbrl/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/data/dataset.py -------------------------------------------------------------------------------- /dbrl/data/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/data/process.py -------------------------------------------------------------------------------- /dbrl/data/session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/data/session.py -------------------------------------------------------------------------------- /dbrl/data/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/data/split.py -------------------------------------------------------------------------------- /dbrl/evaluate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/evaluate/__init__.py -------------------------------------------------------------------------------- /dbrl/evaluate/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/evaluate/evaluate.py -------------------------------------------------------------------------------- /dbrl/evaluate/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/evaluate/metrics.py -------------------------------------------------------------------------------- /dbrl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/models/__init__.py -------------------------------------------------------------------------------- /dbrl/models/bcq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/models/bcq.py -------------------------------------------------------------------------------- /dbrl/models/ddpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/models/ddpg.py -------------------------------------------------------------------------------- /dbrl/models/dssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/models/dssm.py -------------------------------------------------------------------------------- /dbrl/models/youtube_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/models/youtube_topk.py -------------------------------------------------------------------------------- /dbrl/network/__init__.py: -------------------------------------------------------------------------------- 1 | from .net import * 2 | 3 | -------------------------------------------------------------------------------- /dbrl/network/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/network/net.py -------------------------------------------------------------------------------- /dbrl/resources/aa: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dbrl/run_bcq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/run_bcq.py -------------------------------------------------------------------------------- /dbrl/run_ddpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/run_ddpg.py -------------------------------------------------------------------------------- /dbrl/run_prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/run_prepare_data.py -------------------------------------------------------------------------------- /dbrl/run_pretrain_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/run_pretrain_embeddings.py -------------------------------------------------------------------------------- /dbrl/run_reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/run_reinforce.py -------------------------------------------------------------------------------- /dbrl/serialization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/serialization/__init__.py -------------------------------------------------------------------------------- /dbrl/serialization/serialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/serialization/serialize.py -------------------------------------------------------------------------------- /dbrl/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/trainer/__init__.py -------------------------------------------------------------------------------- /dbrl/trainer/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/trainer/pretrain.py -------------------------------------------------------------------------------- /dbrl/trainer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/trainer/train.py -------------------------------------------------------------------------------- /dbrl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/utils/__init__.py -------------------------------------------------------------------------------- /dbrl/utils/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/utils/info.py -------------------------------------------------------------------------------- /dbrl/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/utils/misc.py -------------------------------------------------------------------------------- /dbrl/utils/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/utils/params.py -------------------------------------------------------------------------------- /dbrl/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/dbrl/utils/sampling.py -------------------------------------------------------------------------------- /pic/readme2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/pic/readme2.png -------------------------------------------------------------------------------- /pic/readme5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/massquantity/DBRL/HEAD/pic/readme5.png --------------------------------------------------------------------------------