Commit graph

20 commits

Author SHA1 Message Date
abigailt
367cae679b ModelOutputType is now a Flag instead of regular enum. Combinations of the base flags are provided for all of the previous output types for convenience. All checks in the code now use the basic flags and not the complex types.
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-07-03 13:29:37 +03:00
abigailt
2895b40f05 Addressing review comments
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-06-19 11:17:17 +03:00
abigailt
a4816878f9 Formatting
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-05-02 17:04:34 +03:00
abigailt
a8ec87f922 Add tests for single label binary pytorch models
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-05-02 14:46:08 +03:00
abigailt
8b8b461143 Support for multi-label logits/probabilities
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-03-17 11:49:05 +02:00
abigailt
5e19d4ae27 New model output types + single implementation of score method that supports multiple output types. Existing tests pass. Still need more tests for new types.
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-03-07 19:00:09 +02:00
abigailt
79534b69db Initial support+test for pytorch multi-label binary classifier
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-02-19 14:16:03 +02:00
abigailgold
d52fcd0041
Formatting (#68)
Fix most flake/lint errors and ignore a few others

Signed-off-by: abigailt <abigailt@il.ibm.com>
2022-12-25 15:13:57 +02:00
abigailt
64038f76f9 Merge with main 2022-08-01 18:12:03 +03:00
olasaadi
74ce92acc4 fix 2022-07-26 18:37:44 +03:00
Ron Shmelkin
c77e34e373
update pytorch wrapper to use torch loaders
fix tests
and dataset style
2022-07-24 14:31:47 +03:00
olasaadi
6f69f5557b fix bug 2022-07-20 18:29:48 +03:00
olasaadi
3bf26b67d2 fix 2022-07-20 17:36:00 +03:00
olasaadi
4973fbebc6 fix 2022-07-19 21:16:39 +03:00
olasaadi
21cba95a28 fix 2022-06-06 14:32:34 +03:00
olasaadi
c954f53ad7 fix 2022-06-06 14:02:40 +03:00
olasaadi
302d0c4b8c update 2022-06-02 15:25:07 +03:00
olasaadi
a3fb68fb56 update 2022-05-30 12:52:32 +03:00
olasaadi
023f8764da update 2022-05-30 11:51:22 +03:00
olasaadi
59d8b16bb4 fix 2022-05-23 12:49:38 +03:00