Commit Graph

10 Commits

Author SHA1 Message Date
Francois Chollet
cbc7538a0e Merge branch 'main' of github.com:keras-team/keras-core 2023-04-23 14:02:44 -07:00
Francois Chollet
69d93beb0b Progress on serialization 2023-04-23 14:01:46 -07:00
Aakash Kumar Nain
b3409e383c Add Hinge losses (#31)
* add hinge, squared_hine, and categorical hinge losses

* add tests for hinge, squared_hinge, and categorical hinge losses

* fix keyword argument for jax cond

* format docstrings for hinge losses

* format hinge losses tests

* fix linting errors in activations init and test_case py files

* fix bug in reduce_values function

* rename _transform_binary_labels to convert_binary_labels_to_hinge

* fix casting of  values in  class

* convert shape values to tensor

* add hinge metrics and the corresponding tests

* revert formatting in activations init
2023-04-24 02:30:11 +05:30
Francois Chollet
89b2ea5977 Add fully operational JAX trainer. 2023-04-18 14:49:38 -07:00
Francois Chollet
dc0f3467fc Add draft of JAX trainer 2023-04-17 21:26:04 -07:00
Francois Chollet
f4108a9e9d Add main loss tracking in TF trainer 2023-04-17 17:23:53 -07:00
Francois Chollet
e363fb2b2d Misc fixes. 2023-04-17 15:41:48 -07:00
Francois Chollet
90df7ef9fc Format codebase 2023-04-16 13:30:21 -07:00
Francois Chollet
74c043bdba Fix metric reduction (?) 2023-04-16 12:55:04 -07:00
Francois Chollet
8fa442ccbe Progress on compile utils 2023-04-16 12:21:29 -07:00