diff --git a/docs/carte_ai.data.html b/docs/carte_ai.data.html index 948b98a..58f9b0e 100644 --- a/docs/carte_ai.data.html +++ b/docs/carte_ai.data.html @@ -26,6 +26,8 @@ padding: 2px 4px; border-radius: 3px; } + + @@ -67,32 +69,77 @@
Load and split the Spotify dataset.
-This dataset contains information on over 600,000 Spotify tracks, including audio features and popularity metrics.
-Variables:
-Examples:
-from carte_ai.data.load_data import spotify
-df = spotify()
-print(df.head())
- Load and explore the Spotify dataset, which contains detailed information about over 600,000 Spotify tracks, including audio features, popularity metrics, and genres.
+This dataset can be used for:
+Variables:
+Example Usage:
+
+
+ from carte_ai.data.load_data import *
+
+
+ num_train = 128 # Example: set the number of training groups/entities
+
+ random_state = 1 # Set a random seed for reproducibility
+
+
+ X_train, X_test, y_train, y_test = spotify(num_train, random_state)
+
+
+ # Print dataset shapes
+
+ print("Spotify dataset:", X_train.shape, X_test.shape)
+
+
+
+
+
+
+ from carte_ai.data.load_data import *
+
+
+ num_train = 128 # Example: set the number of training groups/entities
+
+ random_state = 1 # Set a random seed for reproducibility
+
+
+ X_train, X_test, y_train, y_test = wina_pl(num_train, random_state)
+
+
+ # Print dataset shapes
+
+ print("Wina Poland dataset:", X_train.shape, X_test.shape)
+
+
+
+