parsnip.ParsnipModel.fit

ParsnipModel.fit(dataset, max_epochs=1000, augment=True, test_dataset=None)

Fit the model to a dataset

Parameters:

  • dataset (Dataset) – Dataset to fit to

  • max_epochs (int, optional) – Maximum number of epochs, by default 1000

  • augment (bool, optional) – Whether to use augmentation, by default True

  • test_dataset (Dataset, optional) – Test dataset that will be scored at the end of each epoch, by default None