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 |
|