https://github.com/nixtla logo
m

marah othman

05/03/2023, 12:29 PM
hello i am using neuralforcast and i want to know if there way to add earlystopping or the library contains that?
k

Kin Gtz. Olivares

05/03/2023, 1:01 PM
Most models have early stopping by default, you can control its definition through ‘early_stop_patience_steps’, ‘valid_loss’, ‘valid_check_steps’. Here is the documentation of the NHITS model. https://nixtla.github.io/neuralforecast/models.nhits.html#nhits Additionally if you are interested in the source code check this example of the inner workings. https://github.com/Nixtla/neuralforecast/blob/main/neuralforecast/common/_base_windows.py#L126
👍 1