carte_ai.data package

Submodules

carte_ai.data.load_data module

carte_ai.data.load_data.load_parquet_config(data_name)

Helper function to load parquet and config JSON for a given dataset.

carte_ai.data.load_data.set_split(data, config_data, num_train, random_state=42)

Helper function to split dataset into train and test.

carte_ai.data.load_data.spotify(num_train, random_state=42)

Load and split Spotify dataset.

carte_ai.data.load_data.wina_pl(num_train, random_state=42)

Load and split Wina_PL dataset.

carte_ai.data.load_data.wine_dot_com_prices(num_train, random_state=42)

Load and split Wine.com prices dataset.

carte_ai.data.load_data.wine_vivino_price(num_train, random_state=42)

Load and split Vivino wine prices dataset.

Module contents