├── .gitignore ├── LICENSE.txt ├── README.md ├── base ├── BaseDataset.py ├── BaseModel.py ├── __init__.py └── __pycache__ │ ├── BaseDataset.cpython-37.pyc │ ├── BaseDataset.cpython-38.pyc │ ├── BaseModel.cpython-37.pyc │ ├── BaseModel.cpython-38.pyc │ ├── __init__.cpython-37.pyc │ └── __init__.cpython-38.pyc ├── data └── marco-passage │ ├── doc_id_full_sorted.pkl │ ├── doc_id_small_sorted.pkl │ ├── doc_id_valid.pkl │ ├── doc_id_valid_sorted.pkl │ ├── qrels.train.tsv │ ├── queries.train.tsv │ └── test_data │ ├── qrels_dev.tsv │ ├── queries_dev.featpkl │ ├── queries_dev.pkl │ ├── queries_dev.tsv │ └── rels_dev.txt ├── dataloader ├── DataBatcher.py ├── DataLoader.py ├── Dataset.py ├── __init__.py └── __pycache__ │ ├── DataBatcher.cpython-37.pyc │ ├── DataBatcher.cpython-38.pyc │ ├── Dataset.cpython-37.pyc │ ├── Dataset.cpython-38.pyc │ ├── Preprocess.cpython-37.pyc │ ├── Preprocess.cpython-38.pyc │ ├── __init__.cpython-37.pyc │ └── __init__.cpython-38.pyc ├── eval.py ├── evaluation ├── Evaluator.py ├── __init__.py └── __pycache__ │ ├── Evaluator.cpython-37.pyc │ ├── Evaluator.cpython-38.pyc │ ├── __init__.cpython-37.pyc │ └── __init__.cpython-38.pyc ├── experiment ├── EarlyStop.py ├── Experiment.py ├── __init__.py └── __pycache__ │ ├── EarlyStop.cpython-37.pyc │ ├── EarlyStop.cpython-38.pyc │ ├── Experiment.cpython-37.pyc │ ├── Experiment.cpython-38.pyc │ ├── __init__.cpython-37.pyc │ └── __init__.cpython-38.pyc ├── generate_and_eval.py ├── main_config.cfg ├── model ├── SpaDE.py ├── SpaDE_cotraining.py ├── __init__.py └── __pycache__ │ ├── LASER.cpython-37.pyc │ ├── LASER.cpython-38.pyc │ ├── LASER_all.cpython-37.pyc │ ├── LASER_all.cpython-38.pyc │ ├── LASER_context.cpython-37.pyc │ ├── LASER_context.cpython-38.pyc │ ├── LASER_cotraining.cpython-37.pyc │ ├── LASER_cotraining.cpython-38.pyc │ ├── LASER_cotraining_joint.cpython-37.pyc │ ├── LASER_frozen.cpython-37.pyc │ ├── LASER_frozen.cpython-38.pyc │ ├── LASERcatQ.cpython-37.pyc │ ├── LASERcatQ.cpython-38.pyc │ ├── SPLADE_doc.cpython-37.pyc │ ├── SPLADE_doc.cpython-38.pyc │ ├── SpaDE.cpython-37.pyc │ ├── SpaDE_cotraining.cpython-37.pyc │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── gating_model.cpython-37.pyc │ ├── gating_model.cpython-38.pyc │ ├── gating_model2.cpython-37.pyc │ ├── gating_model2.cpython-38.pyc │ ├── gating_model3.cpython-37.pyc │ ├── gating_model3.cpython-38.pyc │ ├── gating_model4.cpython-37.pyc │ └── gating_model4.cpython-38.pyc ├── model_config ├── SpaDE.cfg └── SpaDE_cotraining.cfg ├── requirements.txt ├── saves ├── SpaDE_expand │ ├── SpaDE.cfg │ └── main_config.cfg └── SpaDE_weight │ ├── SpaDE.cfg │ └── main_config.cfg ├── scripts └── run_train.sh ├── train.py └── utils ├── Config.py ├── Constant.py ├── Logger.py ├── ResultTable.py ├── Statistics.py ├── Tool.py ├── __init__.py └── __pycache__ ├── Config.cpython-37.pyc ├── Config.cpython-38.pyc ├── Constant.cpython-37.pyc ├── Constant.cpython-38.pyc ├── Logger.cpython-37.pyc ├── Logger.cpython-38.pyc ├── ResultTable.cpython-37.pyc ├── ResultTable.cpython-38.pyc ├── Tool.cpython-37.pyc ├── Tool.cpython-38.pyc ├── __init__.cpython-37.pyc └── __init__.cpython-38.pyc /.gitignore: -------------------------------------------------------------------------------- 1 | saves/ 2 | -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/README.md -------------------------------------------------------------------------------- /base/BaseDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/BaseDataset.py -------------------------------------------------------------------------------- /base/BaseModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/BaseModel.py -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/__pycache__/BaseDataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/BaseDataset.cpython-37.pyc -------------------------------------------------------------------------------- /base/__pycache__/BaseDataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/BaseDataset.cpython-38.pyc -------------------------------------------------------------------------------- /base/__pycache__/BaseModel.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/BaseModel.cpython-37.pyc -------------------------------------------------------------------------------- /base/__pycache__/BaseModel.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/BaseModel.cpython-38.pyc -------------------------------------------------------------------------------- /base/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /base/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/base/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /data/marco-passage/doc_id_full_sorted.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/doc_id_full_sorted.pkl -------------------------------------------------------------------------------- /data/marco-passage/doc_id_small_sorted.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/doc_id_small_sorted.pkl -------------------------------------------------------------------------------- /data/marco-passage/doc_id_valid.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/doc_id_valid.pkl -------------------------------------------------------------------------------- /data/marco-passage/doc_id_valid_sorted.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/doc_id_valid_sorted.pkl -------------------------------------------------------------------------------- /data/marco-passage/qrels.train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/qrels.train.tsv -------------------------------------------------------------------------------- /data/marco-passage/queries.train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/queries.train.tsv -------------------------------------------------------------------------------- /data/marco-passage/test_data/qrels_dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/test_data/qrels_dev.tsv -------------------------------------------------------------------------------- /data/marco-passage/test_data/queries_dev.featpkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/test_data/queries_dev.featpkl -------------------------------------------------------------------------------- /data/marco-passage/test_data/queries_dev.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/test_data/queries_dev.pkl -------------------------------------------------------------------------------- /data/marco-passage/test_data/queries_dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/test_data/queries_dev.tsv -------------------------------------------------------------------------------- /data/marco-passage/test_data/rels_dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/data/marco-passage/test_data/rels_dev.txt -------------------------------------------------------------------------------- /dataloader/DataBatcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/DataBatcher.py -------------------------------------------------------------------------------- /dataloader/DataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/DataLoader.py -------------------------------------------------------------------------------- /dataloader/Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/Dataset.py -------------------------------------------------------------------------------- /dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__init__.py -------------------------------------------------------------------------------- /dataloader/__pycache__/DataBatcher.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/DataBatcher.cpython-37.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/DataBatcher.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/DataBatcher.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/Dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/Dataset.cpython-37.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/Dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/Dataset.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/Preprocess.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/Preprocess.cpython-37.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/Preprocess.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/Preprocess.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/dataloader/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/eval.py -------------------------------------------------------------------------------- /evaluation/Evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/Evaluator.py -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/__pycache__/Evaluator.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/__pycache__/Evaluator.cpython-37.pyc -------------------------------------------------------------------------------- /evaluation/__pycache__/Evaluator.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/__pycache__/Evaluator.cpython-38.pyc -------------------------------------------------------------------------------- /evaluation/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /evaluation/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/evaluation/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /experiment/EarlyStop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/EarlyStop.py -------------------------------------------------------------------------------- /experiment/Experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/Experiment.py -------------------------------------------------------------------------------- /experiment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__init__.py -------------------------------------------------------------------------------- /experiment/__pycache__/EarlyStop.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/EarlyStop.cpython-37.pyc -------------------------------------------------------------------------------- /experiment/__pycache__/EarlyStop.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/EarlyStop.cpython-38.pyc -------------------------------------------------------------------------------- /experiment/__pycache__/Experiment.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/Experiment.cpython-37.pyc -------------------------------------------------------------------------------- /experiment/__pycache__/Experiment.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/Experiment.cpython-38.pyc -------------------------------------------------------------------------------- /experiment/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /experiment/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/experiment/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /generate_and_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/generate_and_eval.py -------------------------------------------------------------------------------- /main_config.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/main_config.cfg -------------------------------------------------------------------------------- /model/SpaDE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/SpaDE.py -------------------------------------------------------------------------------- /model/SpaDE_cotraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/SpaDE_cotraining.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/__pycache__/LASER.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_all.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_all.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_all.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_all.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_context.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_context.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_context.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_context.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_cotraining.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_cotraining.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_cotraining.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_cotraining.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_cotraining_joint.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_cotraining_joint.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_frozen.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_frozen.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASER_frozen.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASER_frozen.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASERcatQ.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASERcatQ.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/LASERcatQ.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/LASERcatQ.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/SPLADE_doc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/SPLADE_doc.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/SPLADE_doc.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/SPLADE_doc.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/SpaDE.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/SpaDE.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/SpaDE_cotraining.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/SpaDE_cotraining.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model2.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model2.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model2.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model3.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model3.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model3.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model3.cpython-38.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model4.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model4.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/gating_model4.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model/__pycache__/gating_model4.cpython-38.pyc -------------------------------------------------------------------------------- /model_config/SpaDE.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model_config/SpaDE.cfg -------------------------------------------------------------------------------- /model_config/SpaDE_cotraining.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/model_config/SpaDE_cotraining.cfg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==4.25.1 2 | scipy 3 | -------------------------------------------------------------------------------- /saves/SpaDE_expand/SpaDE.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/saves/SpaDE_expand/SpaDE.cfg -------------------------------------------------------------------------------- /saves/SpaDE_expand/main_config.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/saves/SpaDE_expand/main_config.cfg -------------------------------------------------------------------------------- /saves/SpaDE_weight/SpaDE.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/saves/SpaDE_weight/SpaDE.cfg -------------------------------------------------------------------------------- /saves/SpaDE_weight/main_config.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/saves/SpaDE_weight/main_config.cfg -------------------------------------------------------------------------------- /scripts/run_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/scripts/run_train.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/train.py -------------------------------------------------------------------------------- /utils/Config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/Config.py -------------------------------------------------------------------------------- /utils/Constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/Constant.py -------------------------------------------------------------------------------- /utils/Logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/Logger.py -------------------------------------------------------------------------------- /utils/ResultTable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/ResultTable.py -------------------------------------------------------------------------------- /utils/Statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/Statistics.py -------------------------------------------------------------------------------- /utils/Tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/Tool.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/Config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Config.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Config.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Config.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Constant.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Constant.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Constant.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Constant.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Logger.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Logger.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/ResultTable.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/ResultTable.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/ResultTable.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/ResultTable.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Tool.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Tool.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/Tool.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/Tool.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eunseongc/SpaDE/HEAD/utils/__pycache__/__init__.cpython-38.pyc --------------------------------------------------------------------------------