Arafath10 commited on
Commit
58941b3
1 Parent(s): 91039d5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -2
main.py CHANGED
@@ -11,12 +11,17 @@ from sklearn.preprocessing import StandardScaler
11
  import json
12
  import pickle
13
  from sklearn.pipeline import Pipeline
 
14
 
15
  # instantiate flask
16
  app = flask.Flask(__name__)
17
 
18
- # load the model
19
- model = load_model('final_model.h5')
 
 
 
 
20
 
21
  holidays_tt = ["2020-01-01",
22
  "2020-01-15",
 
11
  import json
12
  import pickle
13
  from sklearn.pipeline import Pipeline
14
+ from keras.optimizers import Adam
15
 
16
  # instantiate flask
17
  app = flask.Flask(__name__)
18
 
19
+ # Load the model without compilation
20
+ model = load_model('final_model.h5', compile=False)
21
+
22
+ # Recompile the model with the appropriate optimizer
23
+ optimizer = Adam(learning_rate=0.001) # Adjust learning_rate as needed
24
+ model.compile(optimizer=optimizer, loss='mean_squared_error') # Adjust the loss function as needed
25
 
26
  holidays_tt = ["2020-01-01",
27
  "2020-01-15",