├── README.md ├── bin ├── define.sh └── read_ini.sh ├── data └── .keep ├── environment.yml ├── exclude ├── logs └── .keep ├── rsync.sh ├── scripts ├── __init__.py ├── flen.py ├── flend.py └── jarvis │ ├── __init__.py │ ├── __version__.py │ ├── api │ └── __init__.py │ ├── callbacks.py │ ├── core │ ├── __init__.py │ ├── model.py │ └── utils.py │ ├── data │ ├── __init__.py │ └── ops │ │ ├── __init__.py │ │ └── dataset.py │ ├── initializer.py │ ├── layers │ ├── __init__.py │ ├── bi_interaction.py │ ├── field_wise_bi_interaction.py │ └── fully_connects.py │ ├── losses.py │ ├── metrics.py │ ├── models │ ├── __init__.py │ ├── flen.py │ └── flend.py │ ├── optimizer │ └── __init__.py │ ├── regularizer.py │ ├── tools │ └── inspect_checkpoint.py │ └── utils │ ├── __init__.py │ ├── dataset_utils.py │ ├── flags │ ├── __init__.py │ └── common_flags.py │ └── misc.py └── tools ├── .keep ├── dataset ├── __init__.py └── avazu.py ├── nvidia-smi.sh └── tfrecord_reader.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/README.md -------------------------------------------------------------------------------- /bin/define.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/bin/define.sh -------------------------------------------------------------------------------- /bin/read_ini.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/bin/read_ini.sh -------------------------------------------------------------------------------- /data/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/environment.yml -------------------------------------------------------------------------------- /exclude: -------------------------------------------------------------------------------- 1 | .git/ 2 | .gitignore 3 | rsync.sh 4 | exclude 5 | -------------------------------------------------------------------------------- /logs/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rsync.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/rsync.sh -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/flen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/flen.py -------------------------------------------------------------------------------- /scripts/flend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/flend.py -------------------------------------------------------------------------------- /scripts/jarvis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/__version__.py -------------------------------------------------------------------------------- /scripts/jarvis/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/callbacks.py -------------------------------------------------------------------------------- /scripts/jarvis/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/core/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/core/model.py -------------------------------------------------------------------------------- /scripts/jarvis/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/core/utils.py -------------------------------------------------------------------------------- /scripts/jarvis/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/data/ops/__init__.py: -------------------------------------------------------------------------------- 1 | from dataset import * -------------------------------------------------------------------------------- /scripts/jarvis/data/ops/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/data/ops/dataset.py -------------------------------------------------------------------------------- /scripts/jarvis/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/initializer.py -------------------------------------------------------------------------------- /scripts/jarvis/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/layers/__init__.py -------------------------------------------------------------------------------- /scripts/jarvis/layers/bi_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/layers/bi_interaction.py -------------------------------------------------------------------------------- /scripts/jarvis/layers/field_wise_bi_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/layers/field_wise_bi_interaction.py -------------------------------------------------------------------------------- /scripts/jarvis/layers/fully_connects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/layers/fully_connects.py -------------------------------------------------------------------------------- /scripts/jarvis/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/losses.py -------------------------------------------------------------------------------- /scripts/jarvis/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/metrics.py -------------------------------------------------------------------------------- /scripts/jarvis/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/models/flen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/models/flen.py -------------------------------------------------------------------------------- /scripts/jarvis/models/flend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/models/flend.py -------------------------------------------------------------------------------- /scripts/jarvis/optimizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/regularizer.py -------------------------------------------------------------------------------- /scripts/jarvis/tools/inspect_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/tools/inspect_checkpoint.py -------------------------------------------------------------------------------- /scripts/jarvis/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/utils/dataset_utils.py -------------------------------------------------------------------------------- /scripts/jarvis/utils/flags/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/jarvis/utils/flags/common_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/utils/flags/common_flags.py -------------------------------------------------------------------------------- /scripts/jarvis/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/scripts/jarvis/utils/misc.py -------------------------------------------------------------------------------- /tools/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/dataset/avazu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/tools/dataset/avazu.py -------------------------------------------------------------------------------- /tools/nvidia-smi.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | watch -n 0.1 nvidia-smi -------------------------------------------------------------------------------- /tools/tfrecord_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimetrics/jarvis/HEAD/tools/tfrecord_reader.py --------------------------------------------------------------------------------