lenawilli commited on
Commit
2359af0
·
verified ·
1 Parent(s): f49d247

Update src/recommendation_utils.py

Browse files
Files changed (1) hide show
  1. src/recommendation_utils.py +12 -0
src/recommendation_utils.py CHANGED
@@ -2,6 +2,18 @@ import numpy as np
2
  import pandas as pd
3
  import pickle
4
  from keras.models import load_model
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  # Lade Modelle und Daten
7
  def load_nn_model(path):
 
2
  import pandas as pd
3
  import pickle
4
  from keras.models import load_model
5
+ from keras.models import model_from_json
6
+ from keras.optimizers import Adam
7
+
8
+ def load_nn_model(config_path, weights_path):
9
+ with open(config_path, "r") as f:
10
+ model_json = f.read()
11
+ model = model_from_json(model_json)
12
+ model.load_weights(weights_path)
13
+
14
+ # Gleiche Konfiguration wie im Training:
15
+ model.compile(optimizer=Adam(), loss="mse", metrics=["mse", "mae"])
16
+ return model
17
 
18
  # Lade Modelle und Daten
19
  def load_nn_model(path):