Add note about tensorflow>=2.11 (#738)

cleanlab · Jun 7, 2023 · bf3b56c · bf3b56c
1 parent da65a97
commit bf3b56c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cleanlab/models/keras.py b/cleanlab/models/keras.py
@@ -24,6 +24,8 @@
 This is a good example of making any bespoke neural network compatible with cleanlab.
 
 You must have `Tensorflow 2 installed <https://www.tensorflow.org/install>`_ (only compatible with Python versions >= 3.7).
+This wrapper class is only fully compatible with ``tensorflow<2.11``, if using ``tensorflow>=2.11``, 
+please replace your Optimizer class with the legacy Optimizer `here <https://www.tensorflow.org/api_docs/python/tf/keras/optimizers/legacy/Optimizer>`_.
 
 Tips: