sortvalues 参数修改

52d9d5ee · 张彦钊 · 3bc1226a · 52d9d5ee
Commit 52d9d5ee authored Aug 08, 2018 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

predictDiary.py predictDiary.py +1 -1

No files found.
--- a/predictDiary.py
+++ b/predictDiary.py
@@ -64,7 +64,7 @@ def upload_predict(user_profile, instance):
                                "result/{0}_output.txt".format(user_profile['device_id']), header=None)
    probabilities = probabilities.rename(columns={0: "prob"})
    probabilities["cid"] = instance['cid']
-    probabilities = probabilities.sort_values(by="0", ascending=False)
+    probabilities = probabilities.sort_values(by="prob", ascending=False)

    wrapper_result(probabilities, user_profile['device_id'])