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