TuanScientist commited on
Commit
e4bef95
·
1 Parent(s): 49c8dfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -3,7 +3,6 @@ import pandas as pd
3
  from neuralprophet import NeuralProphet
4
  import io
5
  import warnings
6
- import torch.optim.lr_scheduler as lr_scheduler
7
 
8
  warnings.filterwarnings("ignore", category=UserWarning)
9
 
@@ -15,12 +14,8 @@ df.fillna(method='ffill', inplace=True)
15
  df.dropna(inplace=True)
16
 
17
  class CustomNeuralProphet(NeuralProphet):
18
- def __init__(self, **kwargs):
19
- super().__init__(**kwargs)
20
- self.lr_scheduler = lr_scheduler.ReduceLROnPlateau(self.trainer.optimizers[0], mode='min', factor=0.5, patience=10, verbose=True)
21
-
22
  def lr_scheduler_step(self, metrics):
23
- self.lr_scheduler.step(metrics)
24
 
25
  m = CustomNeuralProphet(
26
  n_forecasts=30,
 
3
  from neuralprophet import NeuralProphet
4
  import io
5
  import warnings
 
6
 
7
  warnings.filterwarnings("ignore", category=UserWarning)
8
 
 
14
  df.dropna(inplace=True)
15
 
16
  class CustomNeuralProphet(NeuralProphet):
 
 
 
 
17
  def lr_scheduler_step(self, metrics):
18
+ pass
19
 
20
  m = CustomNeuralProphet(
21
  n_forecasts=30,