From 353138bd6b570e628255bb47dd68f05a812f42e5 Mon Sep 17 00:00:00 2001 From: gaetanbrison Date: Tue, 3 Dec 2024 11:52:11 +0100 Subject: [PATCH] =?UTF-8?q?new=20data=20doc=20=F0=9F=93=9A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/carte_ai.data.html | 125 +++++++++++++++++++++++++++++----------- 1 file changed, 92 insertions(+), 33 deletions(-) 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:

-
    -
  • track_id: Unique identifier for the track.
    • -
  • artists: Name(s) of the artist(s).
    • -
  • album_name: Name of the album.
    • -
  • track_name: Name of the track.
    • -
  • popularity: Popularity score of the track.
    • -
  • duration_ms: Duration of the track in milliseconds.
    • -
  • danceability: Danceability score (0.0 to 1.0).
    • -
  • energy: Energy score (0.0 to 1.0).
    • -
-

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) +
+
+ +