yzha commited on
Commit
999d570
1 Parent(s): b0eb96e
Files changed (1) hide show
  1. ctc_eval.py +7 -0
ctc_eval.py CHANGED
@@ -83,6 +83,13 @@ class CTC_Eval(evaluate.EvaluationModule):
83
  # TODO: Download external resources if needed
84
  import nltk
85
  nltk.download('stopwords')
 
 
 
 
 
 
 
86
  from ctc_score import StyleTransferScorer, SummarizationScorer, DialogScorer
87
  self.scorer = SummarizationScorer(align='D-cnndm')
88
 
 
83
  # TODO: Download external resources if needed
84
  import nltk
85
  nltk.download('stopwords')
86
+ import subprocess
87
+ import sys
88
+
89
+ def install(package):
90
+ subprocess.check_call([sys.executable, "-m", "pip", "install", package])
91
+
92
+ install('ctc-score')
93
  from ctc_score import StyleTransferScorer, SummarizationScorer, DialogScorer
94
  self.scorer = SummarizationScorer(align='D-cnndm')
95