holisticai.datasets.load_adult
- holisticai.datasets.load_adult(data_home=None, return_X_y=False, as_frame=True)
Load Adult Dataset.
Description
We use the sklearn.datasets fetch_openml function to fetch data from the openml api. Information about the Adult dataset can be obtained by printing load_adult()[‘DESCR’].
Parameters
- data_homestr
The directory to which the data is downloaded. If None, we download to the holisticai_data folder in user home directory.
- return_X_ybool, default=False
If True, returns
(data, target)
instead of a Bunch object. See below for more information about the data and target object.- as_framebool, default=False
If True, the data is a pandas DataFrame including columns with appropriate dtypes (numeric). The target is a pandas DataFrame or Series depending on the number of target columns. If return_X_y is True, then (data, target) will be pandas DataFrames or Series as described below.
Returns
- data
Bunch
Dictionary-like object, with the following attributes. data : {ndarray, dataframe} of shape (1797, 64)
The flattened data matrix. If as_frame=True, data will be a pandas DataFrame.
- target: {ndarray, Series} of shape (1797,)
The classification target. If as_frame=True, target will be a pandas Series.
- frame: DataFrame of shape (1797, 65)
Only present when as_frame=True. DataFrame with data and target.
- feature_names: list
The names of the dataset columns.
- target_names: list
The names of target classes.
- DESCR: str
The full description of the dataset.
- (data, target)tuple if
return_X_y
is True A tuple of two ndarrays by default. The first contains a 2D ndarray of shape (num_rows, num_columns) with each row representing one sample and each column representing the features. The second ndarray of shape (1797) contains the target samples. If as_frame=True, both arrays are pandas objects, i.e. X a dataframe and y a series.