diff --git a/applications/neural_search/recall/in_batch_negative/evaluate.py b/applications/neural_search/recall/in_batch_negative/evaluate.py index 2f3f05a529e1..4a4236220d3b 100644 --- a/applications/neural_search/recall/in_batch_negative/evaluate.py +++ b/applications/neural_search/recall/in_batch_negative/evaluate.py @@ -12,26 +12,22 @@ # See the License for the specific language governing permissions and # limitations under the License. -import os import argparse import numpy as np -from paddlenlp.utils.log import logger import time -# yapf: disable parser = argparse.ArgumentParser() parser.add_argument("--similar_text_pair", type=str, default='', help="The full path of similar pair file") parser.add_argument("--recall_result_file", type=str, default='', help="The full path of recall result file") parser.add_argument("--recall_num", type=int, default=10, - help="Most similair number of doc recalled from corpus per query") + help="Most similar number of doc recalled from corpus per query") args = parser.parse_args() -# yapf: enable def recall(rs, N=10):