google-research / bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.
https://arxiv.org/abs/2004.04696
Apache License 2.0
697 stars 85 forks source link

issue #34

Open jhkd-kevin opened 3 years ago

jhkd-kevin commented 3 years ago

As I follow the command until:

wget https://storage.googleapis.com/bleurt-oss/bleurt-base-128.zip . unzip bleurt-base-128.zip python -m bleurt.score \ -candidate_file=bleurt/test_data/candidates \ -reference_file=bleurt/test_data/references \ -bleurt_checkpoint=bleurt-base-128

I actually finish the step in your program, but as I run Python's API , the code like this:

from bleurt import score

checkpoint = "C:\bleurt-master\bert-base-128" references = ["This is a test."] candidates = ["This is the test."]

scorer = score.BleurtScorer(checkpoint) scores = scorer.score(references=references, candidates=candidates) assert type(scores) == list and len(scores) == 1 print(scores)

It occurred that:AssertionError: Could not find BLEURT checkpoint Cleurt-masteert-base-128