Cache individual preds

wenhui-prudencemed · Apr 23, 2017 · d9f3ea9 · d9f3ea9
1 parent e8f88f1
commit d9f3ea9
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/ecg/predict.py b/ecg/predict.py
@@ -6,15 +6,15 @@
 
 
 @memory.cache
-def get_ensemble_pred_probs(model_paths, x, geo_mean=False):
-    print("Averaging " + str(len(model_paths)) + " model predictions...")
+def get_model_pred_probs(model_path, x):
+    from keras.models import load_model
+    model = load_model(model_path)
+    probs = model.predict(x, verbose=1)
+    return probs
 
-    def get_model_pred_probs(model_path, x):
-        from keras.models import load_model
-        model = load_model(model_path)
-        probs = model.predict(x, verbose=1)
-        return probs
 
+def get_ensemble_pred_probs(model_paths, x, geo_mean=False):
+    print("Averaging " + str(len(model_paths)) + " model predictions...")
     all_model_probs = [get_model_pred_probs(model_path, x)
                        for model_path in model_paths]
     if geo_mean is True: