apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-05-02 16:22:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Create initial version of wrappers for models (#1)

Browse source 
* New wrapper classes for models
This commit is contained in:
ABIGAIL GOLDSTEEN 2022-02-10 15:36:41 +02:00 • committed by GitHub Enterprise
parent 9de078f937
commit b0c6c4d28e
8 changed files with 325 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/utils/models/__init__.py Normal file
View file

@ -0,0 +1,2 @@
from apt.utils.models.model import Model, ModelWithLoss, SingleOutputModel, MultipleOutputModel
from apt.utils.models.sklearn_model import SklearnModel, SklearnClassifier, SklearnRegressor