Merge pull request #18 from tdeboissiere/master

Sort model predictions correctly
fchollet · Oct 31, 2016 · 8d8f54a · 8d8f54a
2 parents 48dcedb + fa518ab
commit 8d8f54a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imagenet_utils.py b/imagenet_utils.py
@@ -42,7 +42,7 @@ def decode_predictions(preds, top=5):
         CLASS_INDEX = json.load(open(fpath))
     results = []
     for pred in preds:
-        top_indices = np.argpartition(pred, -top)[-top:][::-1]
+        top_indices = pred.argsort()[-top:][::-1]
         result = [tuple(CLASS_INDEX[str(i)]) + (pred[i],) for i in top_indices]
         results.append(result)
     return results