TensorFlow Estimators
TensorFlow provides Estimators as higher-level APIs, to provide scalable and production-oriented solutions. They take care of all behind-the-scene activities such as creating computational graphs, initializing the variables, training the model, saving checkpoints, and logging TensorBoard files. TensorFlow provides two types of Estimators:
- Canned Estimators: These are premade Estimators available in the TensorFlow estimator module. These are models in a box; you just pass them the input features and they are ready to use. Some examples are Linear Classifier, Linear Regressor, DNN Classifier, and so on.
- Custom Estimators: Users can also create their own estimators from the models they build in TensorFlow Keras. These are user-defined Estimators.
Before being able to use TensorFlow Estimator let us understand two important components of the Estimator pipeline:
Feature columns
The feature_column
module of TensorFlow 2.0 acts as a bridge between...