load_pre_calculated_prediction#
- load_pre_calculated_prediction(data_size: Optional[int] = 15000) Tuple[ndarray, ndarray][source]#
Load the pre-calculated prediction for the Airbnb NYC 2019 dataset.
- Parameters
- data_sizet.Optional[int] , default: 15000
The number of samples to return. If None, returns all the data.
- Returns
- predictionsTuple(np.ndarray, np.ndarray)
The first element is the pre-calculated prediction for the train set. The second element is the pre-calculated prediction for the test set.