parsnip.ParsnipModel.predict_dataset_augmented
- ParsnipModel.predict_dataset_augmented(dataset, augments=10)
Generate predictions for a dataset with augmentation
This will first generate predictions for the dataset without augmentation, and will then generate predictions for the dataset with augmentation the given number of times. This returns a dataframe in the same format as
predict_dataset, but with the following additional columns: - original_object_id: the original object_id for each augmentation. - augmented: True for augmented light curves, False for original ones.