Update app.py

app.py

@@ -4,7 +4,6 @@ from neuralprophet import NeuralProphet
 import io
 import warnings
 from torch.optim.lr_scheduler import OneCycleLR
-from torch.optim.optimizer import Optimizer
 
 warnings.filterwarnings("ignore", category=UserWarning)
 
@@ -42,14 +41,14 @@ m = CustomNeuralProphet(
 )
 
 # Set the custom LR scheduler
-optimizer = m.trainer.optimizers[0]
+optimizer = m.optimizer
 optimizer.lr_scheduler = {
     "scheduler": OneCycleLR(optimizer, max_lr=0.1, steps_per_epoch=100, epochs=10),
     "interval": "step",
     "frequency": 1,
 }
 
-m.fit(df, freq='D')
+m.fit(df, freq='D', epochs=10, validate_each_epoch=True, valid_p=0.2)
 
 future = m.make_future_dataframe(df, periods=30, n_historic_predictions=True)
 forecast = m.predict(future)