├── .gitignore ├── LICENSE ├── README.md ├── abstract_model.py ├── aux_model ├── TrainDecisionStrain.jl ├── abstract_aux_model.py ├── baselines.py ├── decision_tree_model.py ├── fit_aux_model.jl └── supervised_decision_tree_model.py ├── bamler-mandt-adversarial-neg-sampling-iclr2020.pdf ├── dat ├── .gitattributes ├── Amazon-670K.zip └── Wikipedia-500K.zip ├── dataset.py ├── eval_supervised_aux_model.py ├── evaluate.py ├── load_and_evaluate_model.py ├── main_model ├── abstract_main_model.py ├── complex_model.py ├── distmult_model.py ├── supervised_model.1.py ├── supervised_model.py └── supervised_nce_model.py ├── optimizer.py ├── preprocess-extreme-predicton ├── binarize-features.py ├── binarize-labels.py ├── frequencies.ipynb ├── pca.ipynb └── preprocess-all.sh └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | .ipynb_checkpoints 2 | __pycache__ 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/README.md -------------------------------------------------------------------------------- /abstract_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/abstract_model.py -------------------------------------------------------------------------------- /aux_model/TrainDecisionStrain.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/TrainDecisionStrain.jl -------------------------------------------------------------------------------- /aux_model/abstract_aux_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/abstract_aux_model.py -------------------------------------------------------------------------------- /aux_model/baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/baselines.py -------------------------------------------------------------------------------- /aux_model/decision_tree_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/decision_tree_model.py -------------------------------------------------------------------------------- /aux_model/fit_aux_model.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/fit_aux_model.jl -------------------------------------------------------------------------------- /aux_model/supervised_decision_tree_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/aux_model/supervised_decision_tree_model.py -------------------------------------------------------------------------------- /bamler-mandt-adversarial-neg-sampling-iclr2020.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/bamler-mandt-adversarial-neg-sampling-iclr2020.pdf -------------------------------------------------------------------------------- /dat/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/dat/.gitattributes -------------------------------------------------------------------------------- /dat/Amazon-670K.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/dat/Amazon-670K.zip -------------------------------------------------------------------------------- /dat/Wikipedia-500K.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/dat/Wikipedia-500K.zip -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/dataset.py -------------------------------------------------------------------------------- /eval_supervised_aux_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/eval_supervised_aux_model.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/evaluate.py -------------------------------------------------------------------------------- /load_and_evaluate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/load_and_evaluate_model.py -------------------------------------------------------------------------------- /main_model/abstract_main_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/abstract_main_model.py -------------------------------------------------------------------------------- /main_model/complex_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/complex_model.py -------------------------------------------------------------------------------- /main_model/distmult_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/distmult_model.py -------------------------------------------------------------------------------- /main_model/supervised_model.1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/supervised_model.1.py -------------------------------------------------------------------------------- /main_model/supervised_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/supervised_model.py -------------------------------------------------------------------------------- /main_model/supervised_nce_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/main_model/supervised_nce_model.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/optimizer.py -------------------------------------------------------------------------------- /preprocess-extreme-predicton/binarize-features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/preprocess-extreme-predicton/binarize-features.py -------------------------------------------------------------------------------- /preprocess-extreme-predicton/binarize-labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/preprocess-extreme-predicton/binarize-labels.py -------------------------------------------------------------------------------- /preprocess-extreme-predicton/frequencies.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/preprocess-extreme-predicton/frequencies.ipynb -------------------------------------------------------------------------------- /preprocess-extreme-predicton/pca.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/preprocess-extreme-predicton/pca.ipynb -------------------------------------------------------------------------------- /preprocess-extreme-predicton/preprocess-all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/preprocess-extreme-predicton/preprocess-all.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mandt-lab/adversarial-negative-sampling/HEAD/train.py --------------------------------------------------------------------------------