yangwang825 commited on
Commit
b9e7665
1 Parent(s): 082047a

Update datastats.py

Browse files
Files changed (1) hide show
  1. datastats.py +7 -3
datastats.py CHANGED
@@ -296,8 +296,12 @@ class DataStats(evaluate.Metric):
296
  ):
297
  datastats = DataStatsMetric(n_gram, n_workers, lowercase, tokenize)
298
  results = datastats.evaluate_batch(predictions, references)
 
 
 
 
299
  return {
300
- "coverage": float(results['coverage']),
301
- "density": float(results['density']),
302
- "compression": float(results['compression'])
303
  }
 
296
  ):
297
  datastats = DataStatsMetric(n_gram, n_workers, lowercase, tokenize)
298
  results = datastats.evaluate_batch(predictions, references)
299
+ coverage = float(results['coverage'])
300
+ density = float(results['density'])
301
+ compression = float(results['compression'])
302
+ print(coverage, density, compression)
303
  return {
304
+ "coverage": coverage,
305
+ "density": density,
306
+ "compression": compression
307
  }