diff --git a/tools/simnet/train/tf/README.md b/tools/simnet/train/tf/README.md index df4ceb9..c3bc37b 100755 --- a/tools/simnet/train/tf/README.md +++ b/tools/simnet/train/tf/README.md @@ -74,7 +74,7 @@ python tools/tf_record_writer.py ``` python tf_simnet.py --task train - --task_conf examples/cnn_pointwise.json + --task_conf examples/cnn-pointwise.json ``` **参数说明:** * **task**: 任务类型 ,可选择train或predict。 @@ -138,7 +138,7 @@ python tf_simnet.py ``` python tf_simnet.py --task predict - --task_conf examples/cnn_pointwise.json + --task_conf examples/cnn-pointwise.json ``` **参数说明:** 同模型训练部分 diff --git a/tools/simnet/train/tf/utils/controler.py b/tools/simnet/train/tf/utils/controler.py index ebe5ac3..fdd9997 100644 --- a/tools/simnet/train/tf/utils/controler.py +++ b/tools/simnet/train/tf/utils/controler.py @@ -62,7 +62,7 @@ def run_predict(pred, label, config): sess.close() result_file.close() if mode == "pointwise": - mean_acc = mean_acc / step + mean_acc = mean_acc / step*1.0 print >> sys.stderr, "accuracy: %4.2f" % (mean_acc * 100)