holistic-ai
diff --git a/‎docs/source/getting_started/datasets.rst‎
Lines changed: 3 additions & 3 deletions b/‎docs/source/getting_started/datasets.rst‎
Lines changed: 3 additions & 3 deletions
diff --git a/‎docs/source/getting_started/raw_datasets.csv‎
Lines changed: 14 additions & 13 deletions b/‎docs/source/getting_started/raw_datasets.csv‎
Lines changed: 14 additions & 13 deletions
diff --git a/‎docs/source/reference/datasets.rst‎
Lines changed: 1 addition & 5 deletions b/‎docs/source/reference/datasets.rst‎
Lines changed: 1 addition & 5 deletions
@@ -22,9 +22,9 @@ For example, if we want to load the Adult dataset, we can use the following code
 
 .. code-block:: python
 
-   from holisticai.datasets import load_adult
+    from holisticai.datasets import load_hai_datasets
     
-    data, target = load_adult(return_X_y=True)
+    data, target = load_hai_datasets(dataset_name="adult")
 
 .. _processed_datasets:
 
@@ -49,4 +49,4 @@ You can use this processed datasets using the function load_dataset from holisti
 
    from holisticai.datasets import load_dataset
     
-    dataset = load_dataset(dataset_name="adult", processed=True, protected_attribute="sex")
+    dataset = load_dataset(dataset_name="adult", preprocessed=True, protected_attribute="sex")
@@ -1,13 +1,14 @@
-Student,load_student,"Student performance dataset, used to predict students' grades based on various features."
-Adult,load_adult,"Adult dataset, used for income classification tasks based on census data."
-Law School,load_law_school,"Law school admission dataset, used to predict law school admission based on various features."
-Last FM,load_last_fm,"Last FM dataset, used for music recommendation tasks based on user listening history."
-US Crime,load_us_crime,"US Crime dataset, used to analyze and predict crime rates based on various socio-economic factors."
-Heart,load_heart,"Heart failure clinical records dataset, used to predict heart failure events based on clinical features."
-German Credit,load_german_credit,"German Credit dataset, used for credit risk classification based on financial and demographic factors."
-Census KDD,load_census_kdd,"Census KDD dataset, used for income prediction based on detailed census data."
-Bank Marketing,load_bank_marketing,"Bank Marketing dataset, used to predict whether a client will subscribe to a term deposit based on direct marketing campaigns."
-Compas,load_compas_two_year_recid/load_compas_is_recid,"COMPAS dataset, used for recidivism prediction and analyzing bias in criminal justice algorithms."
-Diabetes,load_diabetes,"Diabetes dataset, used for predicting the onset of diabetes based on health features."
-ACS Income,load_acsincome,"American Community Survey Income dataset, used to predict income levels based on demographic and socioeconomic data."
-ACS Public Coverage,load_acspublic,"American Community Survey Public Coverage dataset, used to analyze public health insurance coverage based on demographic data."
+Student,load_hai_dataset("student"),"Student performance dataset, used to predict students' grades based on various features."
+Adult,load_hai_dataset("adult"),"Adult dataset, used for income classification tasks based on census data."
+Law School,load_hai_dataset("law_school"),"Law school admission dataset, used to predict law school admission based on various features."
+Last FM,load_hai_dataset("lastfm"),"Last FM dataset, used for music recommendation tasks based on user listening history."
+US Crime,load_hai_dataset("us_crime"),"US Crime dataset, used to analyze and predict crime rates based on various socio-economic factors."
+Clinical Records,load_hai_dataset("clinical_records"),"Heart failure clinical records dataset, used to predict heart failure events based on clinical features."
+German Credit,load_hai_dataset("german_credit"),"German Credit dataset, used for credit risk classification based on financial and demographic factors."
+Census KDD,load_hai_dataset("census_kdd"),"Census KDD dataset, used for income prediction based on detailed census data."
+Bank Marketing,load_hai_dataset("bank_marketing"),"Bank Marketing dataset, used to predict whether a client will subscribe to a term deposit based on direct marketing campaigns."
+Compas,load_hai_dataset("compas_two_year_recid"),"COMPAS dataset, used for recidivism prediction and analyzing bias in criminal justice algorithms."
+Compas,load_hai_dataset("compas_is_recid"),"COMPAS dataset, used for recidivism prediction and analyzing bias in criminal justice algorithms."
+Diabetes,load_hai_dataset("diabetes"),"Diabetes dataset, used for predicting the onset of diabetes based on health features."
+ACS Income,load_hai_dataset("acsincome"),"American Community Survey Income dataset, used to predict income levels based on demographic and socioeconomic data."
+ACS Public Coverage,load_hai_dataset("acspublic"),"American Community Survey Public Coverage dataset, used to analyze public health insurance coverage based on demographic data."
@@ -33,8 +33,4 @@
     :template: function.rst
     :toctree: .generated/
 
-    load_adult
-    load_last_fm
-    load_law_school
-    load_student
-    load_us_crime
+    load_hai_datasets