https://github.com/deeprec-ai/estimator
Estimator for DeepRec based on TensorFlow Estimator.
https://github.com/deeprec-ai/estimator
Last synced: 2 months ago
JSON representation
Estimator for DeepRec based on TensorFlow Estimator.
- Host: GitHub
- URL: https://github.com/deeprec-ai/estimator
- Owner: DeepRec-AI
- License: apache-2.0
- Created: 2021-11-15T11:38:02.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2024-02-28T08:47:07.000Z (over 1 year ago)
- Last Synced: 2025-04-05T06:11:14.729Z (3 months ago)
- Language: Python
- Homepage:
- Size: 2.54 MB
- Stars: 5
- Watchers: 0
- Forks: 6
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
# DeepRec Estimator
Estimator for DeepRec is based on TensorFlow Estimator, which is high-level TensorFlow API that greatly simplifies machine learning programming.
Estimators encapsulate training, evaluation, prediction, and exporting for your model.
New features in DeepRec Estimator:
- Support GRPC++ for large scale distributed training in parameter server mode.
- Support StarServer for large scale distributed training in parameter server mode.
## Installation
### Prepare for build
**CPU Dev Docker**
| GCC Version | Python Version | IMAGE |
| ----------- | -------------- | --------------------------------------------------------- |
| 9.4.0 | 3.8.10 | alideeprec/deeprec-build:deeprec-dev-cpu-py38-ubuntu20.04 |
**GPU(cuda11.6) Dev Docker**
| GCC Version | Python Version | CUDA VERSION | IMAGE |
| ----------- | -------------- | ------------ | --------------------------------------------------------------- |
| 9.4.0 | 3.8.10 | CUDA 11.6.2 | alideeprec/deeprec-build:deeprec-dev-gpu-py38-cu116-ubuntu20.04 |
### Build from source
Develop Branch：master, Latest Release Branch: deeprec2302
**Build Package Builder**
```bash
bazel build //tensorflow_estimator/tools/pip_package:build_pip_package
```
**Build Package**
```bash
bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package /tmp/estimator_whl
```
Installing DeepRec will install the native tensorflow-estimator by default, please reinstall the compiled Estimator.
## More details
* [GRPC++](https://github.com/DeepRec-AI/DeepRec/blob/main/docs/docs_en/GRPC%2B%2B.md)
* [StarServer](https://github.com/DeepRec-AI/DeepRec/blob/main/docs/docs_en/StarServer.md)
See TensorFlow Estimator [getting started guide](https://www.tensorflow.org/guide/estimators) for an introduction to the Estimator APIs.