├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── covid_event ├── README.md ├── arch.png ├── c_data.py ├── eval.py ├── event_level_perf.png ├── extra │ ├── args.json │ ├── train.log │ └── unmatched.txt ├── finetune.py ├── finetune_pt.py ├── inference.py ├── mismatches.png ├── predict.py ├── preds │ └── val_preds.json ├── prepare.py ├── results.png ├── submit.py └── subs │ ├── golden │ ├── can_not_test_sol.jsonl │ ├── cure_sol.jsonl │ ├── death_sol.jsonl │ ├── negative_sol.jsonl │ └── positive_sol.jsonl │ ├── post_process.py │ ├── run-1 │ ├── can_not_test.jsonl │ ├── cure.jsonl │ ├── death.jsonl │ ├── negative.jsonl │ └── positive.jsonl │ ├── run-2 │ ├── can_not_test.jsonl │ ├── cure.jsonl │ ├── death.jsonl │ ├── negative.jsonl │ └── positive.jsonl │ ├── run-3 │ ├── can_not_test.jsonl │ ├── cure.jsonl │ ├── death.jsonl │ ├── negative.jsonl │ └── positive.jsonl │ └── testid2candidates.pkl ├── example_bert.py ├── example_t5.py ├── example_trans_t5.py ├── run.py ├── setup.cfg ├── setup.py ├── ttt ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── args.cpython-37.pyc │ ├── evaluators.cpython-37.pyc │ ├── inputs.cpython-37.pyc │ ├── models.cpython-37.pyc │ ├── t2t_trainer.cpython-37.pyc │ └── utils.cpython-37.pyc ├── args.py ├── evaluators.py ├── inputs.py ├── models.py ├── t2t_trainer.py └── utils.py ├── ttt_demo.png ├── ttt_logo.png ├── ttt_notebook.ipynb └── use_tpu_tutorial ├── README.md ├── cls_metric.py ├── covid_data.py ├── images ├── apply_success_email.png ├── second_apply_email.png └── tpu_workflow.png └── run_train_test.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/README.md -------------------------------------------------------------------------------- /covid_event/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/README.md -------------------------------------------------------------------------------- /covid_event/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/arch.png -------------------------------------------------------------------------------- /covid_event/c_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/c_data.py -------------------------------------------------------------------------------- /covid_event/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/eval.py -------------------------------------------------------------------------------- /covid_event/event_level_perf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/event_level_perf.png -------------------------------------------------------------------------------- /covid_event/extra/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/extra/args.json -------------------------------------------------------------------------------- /covid_event/extra/train.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/extra/train.log -------------------------------------------------------------------------------- /covid_event/extra/unmatched.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/extra/unmatched.txt -------------------------------------------------------------------------------- /covid_event/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/finetune.py -------------------------------------------------------------------------------- /covid_event/finetune_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/finetune_pt.py -------------------------------------------------------------------------------- /covid_event/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/inference.py -------------------------------------------------------------------------------- /covid_event/mismatches.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/mismatches.png -------------------------------------------------------------------------------- /covid_event/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/predict.py -------------------------------------------------------------------------------- /covid_event/preds/val_preds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/preds/val_preds.json -------------------------------------------------------------------------------- /covid_event/prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/prepare.py -------------------------------------------------------------------------------- /covid_event/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/results.png -------------------------------------------------------------------------------- /covid_event/submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/submit.py -------------------------------------------------------------------------------- /covid_event/subs/golden/can_not_test_sol.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/golden/can_not_test_sol.jsonl -------------------------------------------------------------------------------- /covid_event/subs/golden/cure_sol.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/golden/cure_sol.jsonl -------------------------------------------------------------------------------- /covid_event/subs/golden/death_sol.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/golden/death_sol.jsonl -------------------------------------------------------------------------------- /covid_event/subs/golden/negative_sol.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/golden/negative_sol.jsonl -------------------------------------------------------------------------------- /covid_event/subs/golden/positive_sol.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/golden/positive_sol.jsonl -------------------------------------------------------------------------------- /covid_event/subs/post_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/post_process.py -------------------------------------------------------------------------------- /covid_event/subs/run-1/can_not_test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-1/can_not_test.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-1/cure.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-1/cure.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-1/death.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-1/death.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-1/negative.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-1/negative.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-1/positive.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-1/positive.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-2/can_not_test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-2/can_not_test.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-2/cure.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-2/cure.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-2/death.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-2/death.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-2/negative.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-2/negative.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-2/positive.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-2/positive.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-3/can_not_test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-3/can_not_test.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-3/cure.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-3/cure.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-3/death.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-3/death.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-3/negative.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-3/negative.jsonl -------------------------------------------------------------------------------- /covid_event/subs/run-3/positive.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/run-3/positive.jsonl -------------------------------------------------------------------------------- /covid_event/subs/testid2candidates.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/covid_event/subs/testid2candidates.pkl -------------------------------------------------------------------------------- /example_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/example_bert.py -------------------------------------------------------------------------------- /example_t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/example_t5.py -------------------------------------------------------------------------------- /example_trans_t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/example_trans_t5.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/run.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [metadata] 2 | description-file = README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/setup.py -------------------------------------------------------------------------------- /ttt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__init__.py -------------------------------------------------------------------------------- /ttt/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/args.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/args.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/evaluators.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/evaluators.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/inputs.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/inputs.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/models.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/models.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/t2t_trainer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/t2t_trainer.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /ttt/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/args.py -------------------------------------------------------------------------------- /ttt/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/evaluators.py -------------------------------------------------------------------------------- /ttt/inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/inputs.py -------------------------------------------------------------------------------- /ttt/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/models.py -------------------------------------------------------------------------------- /ttt/t2t_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/t2t_trainer.py -------------------------------------------------------------------------------- /ttt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt/utils.py -------------------------------------------------------------------------------- /ttt_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt_demo.png -------------------------------------------------------------------------------- /ttt_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt_logo.png -------------------------------------------------------------------------------- /ttt_notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/ttt_notebook.ipynb -------------------------------------------------------------------------------- /use_tpu_tutorial/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/README.md -------------------------------------------------------------------------------- /use_tpu_tutorial/cls_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/cls_metric.py -------------------------------------------------------------------------------- /use_tpu_tutorial/covid_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/covid_data.py -------------------------------------------------------------------------------- /use_tpu_tutorial/images/apply_success_email.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/images/apply_success_email.png -------------------------------------------------------------------------------- /use_tpu_tutorial/images/second_apply_email.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/images/second_apply_email.png -------------------------------------------------------------------------------- /use_tpu_tutorial/images/tpu_workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/images/tpu_workflow.png -------------------------------------------------------------------------------- /use_tpu_tutorial/run_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangcongcong123/ttt/HEAD/use_tpu_tutorial/run_train_test.py --------------------------------------------------------------------------------