Use the NGC CLI to download:
Copied!
Resource
PyTorch codebase for training and using NCF model
Use the NGC CLI to download:
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
.dockerignore | 10 B | January 31, 2023 UTC | |
convert_test.py | 6.99 KB | January 31, 2023 UTC | |
convert.py | 8.3 KB | January 31, 2023 UTC | |
dataloading.py | 12.91 KB | January 31, 2023 UTC | |
Dockerfile | 868 B | January 31, 2023 UTC | |
feature_spec.py | 1.9 KB | January 31, 2023 UTC | |
full_test_suite.md | 503 B | January 31, 2023 UTC | |
inference.py | 4.66 KB | January 31, 2023 UTC | |
LICENSE.md | 11.12 KB | January 31, 2023 UTC | |
load.py | 3.75 KB | January 31, 2023 UTC | |
ncf.py | 17.39 KB | January 31, 2023 UTC | |
neumf_constants.py | 754 B | January 31, 2023 UTC | |
neumf.py | 3.69 KB | January 31, 2023 UTC | |
prepare_dataset.sh | 3.04 KB | January 31, 2023 UTC | |
README.md | 54.57 KB | January 31, 2023 UTC | |
requirements.txt | 79 B | January 31, 2023 UTC | |
test_cases.sh | 1.7 KB | January 31, 2023 UTC | |
test_dataset.sh | 3.05 KB | January 31, 2023 UTC | |
test_featurespec_correctness.py | 3.29 KB | January 31, 2023 UTC | |
transcode.py | 5.56 KB | January 31, 2023 UTC | |
utils.py | 1.04 KB | January 31, 2023 UTC | |
verify_dataset.sh | 884 B | January 31, 2023 UTC |