def example(): from datasets import Datasets datasets = Datasets() datasets.download() training_data = datasets.load() test_data = datasets.load(test=True) my_xgb_regressor = MyXGBRegressor(datasets) my_xgb_regressor.train(training_data) predictions = my_xgb_regressor.predict(test_data, save=True)