Commit Graph

4 Commits

Author SHA1 Message Date
divyasreepat
3873029035 Added Softmax activation layer (#115)
* added softmax activation layer

* update docstring

* reformatted

* updates after review

* updating missed method name update

* update return statement

* added logsumexp class and tests

* update doc string
2023-05-09 16:03:15 -07:00
Aritra Roy Gosthipaty
4ba06d7653 Add: (Sparse)Top K Categorical Accuracy Metric (#61)
* chore: addingop k categorical accuraracy

* chore: adding top k and in top k

* chore: fixing tests

* chore: y true argmax

* chore: adding sparse top k cat metric

* review coomments
2023-05-01 23:47:26 +05:30
Ian Stenbit
a503324861 Add precision metric (#60)
* Add precision metric

* Implement tf nan_to_num

* Add epsilon to divisor

* Formatting
2023-04-30 10:10:07 -06:00
Ian Stenbit
d6bcc56001 First confusion metric (FalsePositives) (#30)
* Added confusion metrics -- still using TF ops

* Fixed structure + tests pass for TF (still need to port to multi-backend)

* Got rid of most tf deps, still a few more to go

* Full removal of TF. Tests pass for both Jax and TF

* Full removal of TF. Tests pass for both Jax and TF

* Formatting

* Formatting

* Review comments

* More review comments + formatting
2023-04-22 10:10:44 -06:00