Commit graph

33 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
a4816878f9 Formatting
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-05-02 17:04:34 +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
cb70ca10e6
Fix calculation of score for pytorch models when a single column of probabilities is used for binary classification (#91)
Signed-off-by: abigailt <abigailt@il.ibm.com>
2024-01-21 06:56:57 -05:00
abigailgold
5dce961092
Support 1-hot encoded features in anonymization + fixes related to encoding in minimization (#86)
* Support 1-hot encoded features in anonymization (#72)
* Fix anonymization adult notebook + new notebook to demonstrate anonymization on 1-hot encoded data

* Minimizer: No default encoder, if none provided data is supplied to the model as is. Fix data type of representative values. Fix and add more tests.

Signed-off-by: abigailt <abigailt@il.ibm.com>
2023-10-19 11:48:15 +03:00
abigailgold
8a9ef80146
Increase version to 0.2.0 (#74)
* Remove tensorflow dependency if not using keras model
* Remove xgboost dependency if not using xgboost model
* Documentation updates

Signed-off-by: abigailt <abigailt@il.ibm.com>
2023-05-08 12:50:55 +03: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
44d012857f Add loss and optimizer as properties
Signed-off-by: abigailt <abigailt@il.ibm.com>
2022-10-19 17:29:40 +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
fdc6005fce add validation set 2022-07-22 01:01:45 +03:00
olasaadi
65388da605 fix docstring 2022-07-20 18:39:53 +03:00
olasaadi
c2c7a01078 fix bug 2022-07-20 18:36:58 +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
07e64b1f86 fix 2022-07-04 12:58:35 +03:00
olasaadi
af7d615628 fix 2022-07-04 12:55:58 +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
8de77f9afd update 2022-05-30 11:52:47 +03:00
olasaadi
023f8764da update 2022-05-30 11:51:22 +03:00
olasaadi
8459d6961f fix 2022-05-23 13:35:09 +03:00
olasaadi
59d8b16bb4 fix 2022-05-23 12:49:38 +03:00
olasaadi
019f49861d fix 2022-05-23 12:15:35 +03:00
olasaadi
e0385b0d04 score 2022-05-19 04:40:02 +03:00
olasaadi
7539ca0ead save checkpoints 2022-05-19 04:29:55 +03:00
olasaadi
521c8ce041 fix 2022-05-17 16:57:21 +03:00
olasaadi
f484135d84 add classes and implement some functions 2022-05-08 13:50:58 +03:00