bug fix

gofullthrottle · Aug 5, 2018 · 111beb5 · 111beb5
1 parent 5ecdc50
commit 111beb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/age_estimation/model.py b/age_estimation/model.py
@@ -6,8 +6,8 @@
 
 
 def age_mae(y_true, y_pred):
-    true_age = K.sum(y_true * K.arange(0, 101), axis=-1)
-    pred_age = K.sum(y_pred * K.arange(0, 101), axis=-1)
+    true_age = K.sum(y_true * K.arange(0, 101, dtype="float32"), axis=-1)
+    pred_age = K.sum(y_pred * K.arange(0, 101, dtype="float32"), axis=-1)
     mae = K.mean(K.abs(true_age - pred_age))
     return mae