mistral_v0_2.lib.example_data#
- mistral_v0_2.lib.example_data.load_data(splitting)[source]#
Loads the example data, the Yelp review dataset, and returns the text inputs with corresponding labels for either the training or test split.
- Parameters:
splitting (str) – A string indicating which split of the dataset to load. Expected values are ‘train’ for the training set and any other value will default to loading the test set.
- Returns:
- A tuple containing two lists:
The first list contains the text of the reviews.
The second list contains the corresponding labels as integers.
- Return type:
tuple[list[str], list[int]]