What is the difference between tf.keras and tf.python.keras?

by

From an official TensorFlow dev, shortened (emphasis mine):

The API import is in the root of the package. Any other import is just Python allowing you to access privates with no consideration for good coding practices.

The only way that imports should be are

import tensorflow as tf
tf.keras

We also provide support for from tensorflow.keras import,
though this is brittle and can break as we keep refactoring.
Importing from tensorflow.python or any other modules (including import tensorflow_core) is not supported, and can break unannounced.

Me: To confirm, tf.python.keras is private, intended for development, rather than public use?

Yes, that’s exactly the case. Anything under tf.python is private

This, however, is not the full picture. tf.python remains the only way to access certain functions / classes – e.g., tf.python.framework and tf.python.ops, both used in tf.keras.optimizers. But as per above, this doesn’t become a concern unless you’re “developing” – i.e. writing custom functionality or classes. “Out of box” usage should be fine without ever touching tf.python.

Note this isn’t only a compatibility matter, and the two are not interchangeable “as long as nothing breaks”; for example, tf.keras uses optimizer_v2, which differs substantially from tf.python.keras Optimizer.

Lastly, note that both above links end up in tf.python.keras — not certain, but it appears that tf.keras doesn’t actually exist in TF Github (e.g. nothing references OptimizerV2), but it does merge with TF in tensorflow_core/python/keras/api/_v2 folder when installed locally:

from tensorflow import keras
print(keras.__file__)
from tensorflow.python import keras
print(keras.__file__)

D:\Anaconda\lib\site-packages\tensorflow_core\python\keras\api\_v2\keras\__init__.py
D:\Anaconda\lib\site-packages\tensorflow_core\python\keras\__init__.py

Though both share the python/ folder, they’re not both tf.python – can be verified from their respective __init__.py.

UPDATE: tf.python.keras.optimizers used with tf.python.keras.layers vs tf.keras.optimizers used with tf.keras.layers runs 11.5x slower for a mid-sized model (code). I continue to see former in user code – consider this a note of warning.

More Related Contents:

  1. Keras not training on entire dataset
  2. Could not load dynamic library ‘cudart64_101.dll’ on tensorflow CPU-only installation
  3. Why is TensorFlow 2 much slower than TensorFlow 1?
  4. Tensorflow 2.0 – AttributeError: module ‘tensorflow’ has no attribute ‘Session’
  5. This model has not yet been built error on model.summary()
  6. Keras Dense layer’s input is not flattened
  7. TensorFlow Only running on 1/32 of the Training data provided [duplicate]
  8. How to export Keras .h5 to tensorflow .pb?
  9. Tensorflow – ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type float)
  10. Keras + Tensorflow and Multiprocessing in Python
  11. Proper way to feed time-series data to stateful LSTM?
  12. How to use a Keras RNN model to forecast for future dates or events?
  13. Keras replacing input layer
  14. How does Keras 1d convolution layer work with word embeddings – text classification problem? (Filters, kernel size, and all hyperparameter)
  15. Keras Sequential model with multiple inputs
  16. ValueError at /image/ Tensor Tensor(“activation_5/Softmax:0”, shape=(?, 4), dtype=float32) is not an element of this graph
  17. What do I need K.clear_session() and del model for (Keras with Tensorflow-gpu)?
  18. expected ndim=3, found ndim=2
  19. Passing output of a CNN to BILSTM
  20. How to get other metrics in Tensorflow 2.0 (not only accuracy)?
  21. Why can’t I get reproducible results in Keras even though I set the random seeds?
  22. AttributeError: module ‘tensorflow.python.keras.utils.generic_utils’ has no attribute ‘populate_dict_with_module_objects’
  23. How do I check if keras is using gpu version of tensorflow?
  24. Tensorflow : logits and labels must have the same first dimension
  25. Keras LSTM input dimension setting
  26. How to create a keras layer with a custom gradient in TF2.0?
  27. ‘Tensor’ object has no attribute ‘lower’
  28. Unexpected keyword argument ‘ragged’ in Keras
  29. ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`
  30. Invalid Argument Error / Graph Execution Error

Leave a Comment