├── .DS_Store ├── .gitignore ├── aug └── aug.py ├── call_backs ├── EvalAcc_high_low_emb.py ├── MaskExp.py └── __init__.py ├── conf ├── HCL.yaml ├── NG20.yaml ├── cifar10.yaml └── mnist.yaml ├── dash_main.py ├── data_model ├── __init__.py ├── datamodel_tri_test.py ├── dataset_all.py ├── dataset_bio.py ├── dataset_cifar.py ├── dataset_meta.py ├── dataset_nist.py └── dataset_text.py ├── data_path ├── 20NG.npy └── 20NG_labels.npy ├── eval ├── __init__.py ├── eval_core.py ├── eval_core_base.py └── evaluation.py ├── install_env.sh ├── main.py ├── manifolds ├── __init__.py ├── base.py ├── euclidean.py ├── hyperbolic_project.py ├── hyperboloid.py ├── math_utils.py └── poincare.py ├── model ├── __init__.py ├── nnet.py ├── nnet_new_exp.py ├── nnet_new_exp_mix_in_lat_exp_all.py └── resnet.py ├── readme.md ├── requirements.txt ├── run_wandb.sh └── sweep └── nml4 ├── HCL.yaml ├── NG20.yaml ├── cifar10.yaml └── mnist.yaml /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/.gitignore -------------------------------------------------------------------------------- /aug/aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/aug/aug.py -------------------------------------------------------------------------------- /call_backs/EvalAcc_high_low_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/call_backs/EvalAcc_high_low_emb.py -------------------------------------------------------------------------------- /call_backs/MaskExp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/call_backs/MaskExp.py -------------------------------------------------------------------------------- /call_backs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /conf/HCL.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/conf/HCL.yaml -------------------------------------------------------------------------------- /conf/NG20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/conf/NG20.yaml -------------------------------------------------------------------------------- /conf/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/conf/cifar10.yaml -------------------------------------------------------------------------------- /conf/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/conf/mnist.yaml -------------------------------------------------------------------------------- /dash_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/dash_main.py -------------------------------------------------------------------------------- /data_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_model/datamodel_tri_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/datamodel_tri_test.py -------------------------------------------------------------------------------- /data_model/dataset_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_all.py -------------------------------------------------------------------------------- /data_model/dataset_bio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_bio.py -------------------------------------------------------------------------------- /data_model/dataset_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_cifar.py -------------------------------------------------------------------------------- /data_model/dataset_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_meta.py -------------------------------------------------------------------------------- /data_model/dataset_nist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_nist.py -------------------------------------------------------------------------------- /data_model/dataset_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_model/dataset_text.py -------------------------------------------------------------------------------- /data_path/20NG.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_path/20NG.npy -------------------------------------------------------------------------------- /data_path/20NG_labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/data_path/20NG_labels.npy -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/eval_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/eval/eval_core.py -------------------------------------------------------------------------------- /eval/eval_core_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/eval/eval_core_base.py -------------------------------------------------------------------------------- /eval/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/eval/evaluation.py -------------------------------------------------------------------------------- /install_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/install_env.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/main.py -------------------------------------------------------------------------------- /manifolds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/__init__.py -------------------------------------------------------------------------------- /manifolds/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/base.py -------------------------------------------------------------------------------- /manifolds/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/euclidean.py -------------------------------------------------------------------------------- /manifolds/hyperbolic_project.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/hyperbolic_project.py -------------------------------------------------------------------------------- /manifolds/hyperboloid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/hyperboloid.py -------------------------------------------------------------------------------- /manifolds/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/math_utils.py -------------------------------------------------------------------------------- /manifolds/poincare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/manifolds/poincare.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/nnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/model/nnet.py -------------------------------------------------------------------------------- /model/nnet_new_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/model/nnet_new_exp.py -------------------------------------------------------------------------------- /model/nnet_new_exp_mix_in_lat_exp_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/model/nnet_new_exp_mix_in_lat_exp_all.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/model/resnet.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_wandb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/run_wandb.sh -------------------------------------------------------------------------------- /sweep/nml4/HCL.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/sweep/nml4/HCL.yaml -------------------------------------------------------------------------------- /sweep/nml4/NG20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/sweep/nml4/NG20.yaml -------------------------------------------------------------------------------- /sweep/nml4/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/sweep/nml4/cifar10.yaml -------------------------------------------------------------------------------- /sweep/nml4/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zangzelin/code_dmtme/HEAD/sweep/nml4/mnist.yaml --------------------------------------------------------------------------------