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

update docstring

Browse source
This commit is contained in:
olasaadi 2022-03-28 00:07:04 +03:00
parent b54f0a2382
commit 3bc1341dad
2 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
apt/utils/datasets/datasets.py
View file

@ -162,12 +162,13 @@ class StoredDataset(Dataset):
class ArrayDataset(Dataset):
"""Dataset that is based on x and y arrays (e.g., numpy/pandas/list...)"""
def __init__(self, x: INPUT_DATA_ARRAY_TYPE, y: Optional[INPUT_DATA_ARRAY_TYPE] = None, features_names=None,
**kwargs):
def __init__(self, x: INPUT_DATA_ARRAY_TYPE, y: Optional[INPUT_DATA_ARRAY_TYPE] = None,
features_names: Optional = None, **kwargs):
"""
ArrayDataset constructor.
:param x: collection of data samples
:param y: collection of labels (optional)
:param feature_names: list of str, The feature names, in the order that they appear in the data (optional)
:param kwargs: dataset parameters
"""
self.is_pandas = False