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

Update src/recommendation_utils.py

Browse files
Files changed (1) hide show
  1. src/recommendation_utils.py +1 -5
src/recommendation_utils.py CHANGED
@@ -11,14 +11,10 @@ def load_nn_model(config_path, weights_path):
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):
20
- return load_model(path)
21
-
22
  def load_svd_model(path):
23
  with open(path, "rb") as f:
24
  return pickle.load(f)
 
11
  model = model_from_json(model_json)
12
  model.load_weights(weights_path)
13
 
14
+ # same config as used in training
15
  model.compile(optimizer=Adam(), loss="mse", metrics=["mse", "mae"])
16
  return model
17
 
 
 
 
 
18
  def load_svd_model(path):
19
  with open(path, "rb") as f:
20
  return pickle.load(f)