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 @@

carte_ai.data packagecarte_ai.data.load_data module

+
-
- carte_ai.data.load_data. - spotify() - -
-
-

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())
-
-
+
+ carte_ai.data.load_data. + spotify() + +
+
+

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) +
+
+
+ +
+
+ +
@@ -131,15 +178,27 @@

carte_ai.data.load_data module - from carte_ai.data.load_data import wina_pl - - # Load the dataset - df = wina_pl() - - # Display the first few rows - print(df.head()) + +
+ 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) +
+
+ +