@@ -138,7 +138,7 @@ from fibber.fibber import Fibber
138138arg_dict = {
139139 " use_gpu_id" : 0 ,
140140 " gpt2_gpu_id" : 0 ,
141- " bert_gpu_id " : 0 ,
141+ " transformer_clf_gpu_id " : 0 ,
142142 " ce_gpu_id" : 0 ,
143143 " strategy_gpu_id" : 0 ,
144144 " asrs_block_size" : 3 ,
@@ -147,7 +147,7 @@ arg_dict = {
147147 " asrs_sim_threshold" : 0.95 ,
148148 " asrs_ppl_weight" : 5 ,
149149 " asrs_clf_weight" : 3
150- " asrs_sim_metric" : " CESemanticSimilarityMetric "
150+ " asrs_sim_metric" : " CESimilarityMetric "
151151}
152152
153153# create a fibber object.
@@ -189,29 +189,29 @@ The output is a tuple of (str, list, list).
189189# Evaluation metrics of these 5 paraphrase_list.
190190
191191 {' EditingDistance' : 8 ,
192- ' USESemanticSimilarityMetric ' : 0.9523628950119019 ,
193- ' GloVeSemanticSimilarityMetric ' : 0.9795315341042675 ,
194- ' GPT2GrammarQualityMetric ' : 1.492070198059082 ,
192+ ' USESimilarityMetric ' : 0.9523628950119019 ,
193+ ' GloVeSimilarityMetric ' : 0.9795315341042675 ,
194+ ' GPT2PerplexityMetric ' : 1.492070198059082 ,
195195 ' BertClassifier' : 0 },
196196 {' EditingDistance' : 9 ,
197- ' USESemanticSimilarityMetric ' : 0.9372092485427856 ,
198- ' GloVeSemanticSimilarityMetric ' : 0.9575780832312993 ,
199- ' GPT2GrammarQualityMetric ' : 0.9813404679298401 ,
197+ ' USESimilarityMetric ' : 0.9372092485427856 ,
198+ ' GloVeSimilarityMetric ' : 0.9575780832312993 ,
199+ ' GPT2PerplexityMetric ' : 0.9813404679298401 ,
200200 ' BertClassifier' : 1 },
201201 {' EditingDistance' : 11 ,
202- ' USESemanticSimilarityMetric ' : 0.9265919327735901 ,
203- ' GloVeSemanticSimilarityMetric ' : 0.9710499628056698 ,
204- ' GPT2GrammarQualityMetric ' : 1.325406551361084 ,
202+ ' USESimilarityMetric ' : 0.9265919327735901 ,
203+ ' GloVeSimilarityMetric ' : 0.9710499628056698 ,
204+ ' GPT2PerplexityMetric ' : 1.325406551361084 ,
205205 ' BertClassifier' : 0 },
206206 {' EditingDistance' : 7 ,
207- ' USESemanticSimilarityMetric ' : 0.8913971185684204 ,
208- ' GloVeSemanticSimilarityMetric ' : 0.9800737898362042 ,
209- ' GPT2GrammarQualityMetric ' : 1.2504483461380005 ,
207+ ' USESimilarityMetric ' : 0.8913971185684204 ,
208+ ' GloVeSimilarityMetric ' : 0.9800737898362042 ,
209+ ' GPT2PerplexityMetric ' : 1.2504483461380005 ,
210210 ' BertClassifier' : 1 },
211211 {' EditingDistance' : 8 ,
212- ' USESemanticSimilarityMetric ' : 0.9124080538749695 ,
213- ' GloVeSemanticSimilarityMetric ' : 0.9744155151490856 ,
214- ' GPT2GrammarQualityMetric ' : 1.1626977920532227 ,
212+ ' USESimilarityMetric ' : 0.9124080538749695 ,
213+ ' GloVeSimilarityMetric ' : 0.9744155151490856 ,
214+ ' GPT2PerplexityMetric ' : 1.1626977920532227 ,
215215 ' BertClassifier' : 0 }]
216216```
217217
0 commit comments