Spaces:
Sleeping
Sleeping
Commit
·
b673837
1
Parent(s):
0ad6252
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,14 +44,6 @@ m = CustomNeuralProphet(
|
|
| 44 |
learning_rate=0.03,
|
| 45 |
)
|
| 46 |
|
| 47 |
-
# Set the custom LR scheduler
|
| 48 |
-
m.optimizer = m.model.optimizers[0] # Use the optimizer from the model
|
| 49 |
-
|
| 50 |
-
m.optimizer.lr_scheduler = {
|
| 51 |
-
"scheduler": OneCycleLR(m.optimizer),
|
| 52 |
-
"interval": "step",
|
| 53 |
-
"frequency": 1,
|
| 54 |
-
}
|
| 55 |
|
| 56 |
m.fit(df, freq='D')
|
| 57 |
|
|
|
|
| 44 |
learning_rate=0.03,
|
| 45 |
)
|
| 46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
m.fit(df, freq='D')
|
| 49 |
|