├── .gitignore ├── LICENSE ├── README.md ├── data ├── aliccp │ └── dataset description.md ├── avazu │ └── dataset description.md ├── criteo │ └── dataset description.md ├── movielens-1m │ └── dataset desciption.md └── preprocess.py ├── fs_run.py ├── models ├── basemodel.py ├── config.yaml ├── fs │ ├── adafs.py │ ├── autofield.py │ ├── gbdt.py │ ├── lasso.py │ ├── lpfs.py │ ├── mvfs.py │ ├── no_selection.py │ ├── optfs.py │ ├── optfs_old.py │ ├── permutation.py │ ├── rf.py │ ├── sfs.py │ ├── shark.py │ └── xgb.py ├── layers.py └── rec │ ├── dcn.py │ ├── deepfm.py │ ├── fibinet.py │ ├── fm.py │ ├── mlp.py │ └── widedeep.py ├── nni └── search_spaces │ ├── config.json │ └── fs │ ├── adafs.json │ ├── autofield.json │ ├── gbdt.json │ ├── lasso.json │ └── optfs.json ├── nni_tune.py ├── requirements.txt └── utils ├── datasets.py ├── fs_trainer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/README.md -------------------------------------------------------------------------------- /data/aliccp/dataset description.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/data/aliccp/dataset description.md -------------------------------------------------------------------------------- /data/avazu/dataset description.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/data/avazu/dataset description.md -------------------------------------------------------------------------------- /data/criteo/dataset description.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/data/criteo/dataset description.md -------------------------------------------------------------------------------- /data/movielens-1m/dataset desciption.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/data/movielens-1m/dataset desciption.md -------------------------------------------------------------------------------- /data/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/data/preprocess.py -------------------------------------------------------------------------------- /fs_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/fs_run.py -------------------------------------------------------------------------------- /models/basemodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/basemodel.py -------------------------------------------------------------------------------- /models/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/config.yaml -------------------------------------------------------------------------------- /models/fs/adafs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/adafs.py -------------------------------------------------------------------------------- /models/fs/autofield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/autofield.py -------------------------------------------------------------------------------- /models/fs/gbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/gbdt.py -------------------------------------------------------------------------------- /models/fs/lasso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/lasso.py -------------------------------------------------------------------------------- /models/fs/lpfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/lpfs.py -------------------------------------------------------------------------------- /models/fs/mvfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/mvfs.py -------------------------------------------------------------------------------- /models/fs/no_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/no_selection.py -------------------------------------------------------------------------------- /models/fs/optfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/optfs.py -------------------------------------------------------------------------------- /models/fs/optfs_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/optfs_old.py -------------------------------------------------------------------------------- /models/fs/permutation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/permutation.py -------------------------------------------------------------------------------- /models/fs/rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/rf.py -------------------------------------------------------------------------------- /models/fs/sfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/sfs.py -------------------------------------------------------------------------------- /models/fs/shark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/shark.py -------------------------------------------------------------------------------- /models/fs/xgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/fs/xgb.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/rec/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/dcn.py -------------------------------------------------------------------------------- /models/rec/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/deepfm.py -------------------------------------------------------------------------------- /models/rec/fibinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/fibinet.py -------------------------------------------------------------------------------- /models/rec/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/fm.py -------------------------------------------------------------------------------- /models/rec/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/mlp.py -------------------------------------------------------------------------------- /models/rec/widedeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/models/rec/widedeep.py -------------------------------------------------------------------------------- /nni/search_spaces/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/config.json -------------------------------------------------------------------------------- /nni/search_spaces/fs/adafs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/fs/adafs.json -------------------------------------------------------------------------------- /nni/search_spaces/fs/autofield.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/fs/autofield.json -------------------------------------------------------------------------------- /nni/search_spaces/fs/gbdt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/fs/gbdt.json -------------------------------------------------------------------------------- /nni/search_spaces/fs/lasso.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/fs/lasso.json -------------------------------------------------------------------------------- /nni/search_spaces/fs/optfs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni/search_spaces/fs/optfs.json -------------------------------------------------------------------------------- /nni_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/nni_tune.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/fs_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/utils/fs_trainer.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Applied-Machine-Learning-Lab/ERASE/HEAD/utils/utils.py --------------------------------------------------------------------------------