Skip to content

Commit

Permalink
Update evaluate.py (PaddlePaddle#4553)
Browse files Browse the repository at this point in the history
* Update evaluate.py

* Update evaluate.py

---------

Co-authored-by: Sijun He <sijun.he@hotmail.com>
  • Loading branch information
co63oc and sijunhe authored Jan 29, 2023
1 parent e48a7cb commit 461dcc3
Showing 1 changed file with 1 addition and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 461dcc3

Please sign in to comment.