yangwang825 commited on
Commit
ac41e13
1 Parent(s): 901e7a3

Update datastats.py

Browse files
Files changed (1) hide show
  1. datastats.py +3 -1
datastats.py CHANGED
@@ -293,10 +293,12 @@ class DataStats(evaluate.Metric):
293
  predictions,
294
  references,
295
  n_gram: int = 3,
296
- n_workers: int = 24,
297
  lowercase: bool = False,
298
  tokenize: bool = True
299
  ):
 
 
300
  datastats = DataStatsMetric(n_gram, n_workers, lowercase, tokenize)
301
  results = datastats.evaluate_batch(predictions, references)
302
  coverage = float(results['coverage'])
 
293
  predictions,
294
  references,
295
  n_gram: int = 3,
296
+ n_workers: int = 4,
297
  lowercase: bool = False,
298
  tokenize: bool = True
299
  ):
300
+ logger.info(predictions)
301
+ logger.info(references)
302
  datastats = DataStatsMetric(n_gram, n_workers, lowercase, tokenize)
303
  results = datastats.evaluate_batch(predictions, references)
304
  coverage = float(results['coverage'])