Commit Graph

6 Commits

Author SHA1 Message Date
Chen Qian
752f4b581b Add torch.math.* (#189)
* initial

* Add ops

* fix comments
2023-05-18 18:04:25 -07:00
Chen Qian
59876afd28 Export operations (#176)
* Export operations

* dual export
2023-05-16 20:35:11 -07:00
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
Ian Stenbit
55640456ff Add 4 more Confusion Metrics (#73)
* Add recall

* Add SensitivitySpecificity metrics

* Review comments

* Add missing method from math

* Fix pytest

* Update config dict creation

* Remove sentence fragment

* Review comments

* Add special casing for min/max
2023-05-05 10:53:36 -06: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