STT-tensorflow/tensorflow/python/keras
Thomas O'Malley aed7a7b5e8 Add reset_states method to LossesContainer and MetricsContainer.
PiperOrigin-RevId: 358259524
Change-Id: I875ad8434cccd6ca47fc41f9e3a7328cd2fcecb6
2021-02-18 14:10:41 -08:00
..
api Move mixed precision files out of experimental/ directory. 2020-10-21 15:19:21 -07:00
applications PY2 removal cleanup 2021-01-15 16:48:57 -08:00
benchmarks Disable guitar for keras_cpu_benchmark_test 2021-02-16 16:31:16 -08:00
datasets PY2 removal cleanup 2021-01-15 16:48:57 -08:00
distribute Add test of save/load/serving model under MirroredStrategy. 2021-02-17 17:33:58 -08:00
engine Add reset_states method to LossesContainer and MetricsContainer. 2021-02-18 14:10:41 -08:00
estimator
feature_column Remove the usage of variable_scope._pure_variable_scope() in keras. 2021-01-25 10:31:16 -08:00
initializers Fork the implementation of tf.initializer to keras. 2020-12-28 11:13:42 -08:00
integration_test Update keras/integration test build deps to align with OSS keras. 2021-02-11 11:01:01 -08:00
layers Remove crossed hashing functionality from the hashing layer 2021-02-17 15:02:14 -08:00
legacy_tf_layers Remove redundant use_gpu=True params 2021-02-12 22:16:16 -08:00
mixed_precision Disable keras layer_correctness_test for TFRT. 2021-02-11 10:22:14 -08:00
optimizer_v2 Remove redundant use_gpu=True params 2021-02-12 22:16:16 -08:00
premade PY2 removal cleanup 2021-01-20 17:11:44 -08:00
preprocessing PY2 removal cleanup 2021-01-15 16:48:57 -08:00
protobuf Comment out a number of google-internal targets when copybara-exporting instead of removing them. 2021-01-26 05:47:31 -08:00
saving Remove unneeded input extraction from layer tracing. 2021-02-18 12:21:12 -08:00
tests Update visibility setting for keras/tests to enable PIP package testing. 2021-02-04 12:09:57 -08:00
type PY2 removal cleanup 2021-01-20 17:11:44 -08:00
utils Add test of save/load/serving model under MirroredStrategy. 2021-02-17 17:33:58 -08:00
wrappers PY2 removal cleanup 2021-01-15 16:48:57 -08:00
__init__.py SidecarEvaluator: Create a Keras SidecarEvaluator API for a dedicated evaluator task. 2020-10-29 02:13:29 -07:00
activations_test.py
activations.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
backend_config_test.py
backend_config.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
backend_test.py Merge pull request #45386 from ROCmSoftwarePlatform:google_upstream_rocm_misc_update_201203 2020-12-09 07:38:17 -08:00
backend.py Remove duplicated implementation of keras.backend.ndim 2021-02-08 20:20:15 +01:00
BUILD Fix the issue that test_util isn't allowed in keras tests. 2021-02-09 15:59:51 -08:00
callbacks_test.py Disable a few more Python tests that TFRT doesn't support. 2021-02-09 15:35:23 -08:00
callbacks_v1_test.py
callbacks_v1.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
callbacks.py Internal change 2021-02-16 18:34:49 -08:00
combinations_test.py
combinations.py Remove test combination and annotation for keras_tensor. 2021-01-21 10:17:14 -08:00
constraints_test.py
constraints.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
initializers_test.py Modify some v2 initializers to be able to return a value that corresponds to a partition of the entire value. This is useful for efficiently initializing sharded variables where only a shard of the initial value is necessary at a time. 2020-10-21 18:20:24 -07:00
keras_parameterized_test.py Remove test combination and annotation for keras_tensor. 2021-01-21 10:17:14 -08:00
keras_parameterized.py Remove redundant use_gpu=True params 2021-02-12 22:16:16 -08:00
losses_test.py Merge pull request #47075 from pedro-r-marques:rt-bce 2021-02-17 17:09:16 -08:00
losses.py Merge pull request #47075 from pedro-r-marques:rt-bce 2021-02-17 17:09:16 -08:00
metrics_confusion_matrix_test.py Add support for AUC and PR-AUC for binary classification model outputting logits. 2020-12-03 06:37:07 -08:00
metrics_correctness_test.py Move the LossReduction class from tf to Keras. 2021-01-13 13:40:14 -08:00
metrics_functional_test.py
metrics_test.py Add support for AUC and PR-AUC for binary classification model outputting logits. 2020-12-03 06:37:07 -08:00
metrics.py Remove the usage of with_dependency() in keras code base. 2021-01-25 11:33:56 -08:00
models_test.py Add support for shared objects in keras when (de)serializing. 2021-02-01 12:27:05 -08:00
models.py Add support for shared objects in keras when (de)serializing. 2021-02-01 12:27:05 -08:00
optimizer_v1.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
optimizers_test.py Move V1 optimizer code to a separate file optimizer_v1.py from the generic utils in optimizers.py. 2020-09-24 05:34:47 -07:00
optimizers.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
README.md
regularizers_test.py
regularizers.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00
testing_utils.py [*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:" 2020-12-22 09:24:04 +11:00

Keras is an object-oriented API for defining and training neural networks.

This module contains a pure-TensorFlow implementation of the Keras API, allowing for deep integration with TensorFlow functionality.

See keras.io for complete documentation and user guides.