Upload TestEnsembleQAPipeline
Browse files- test_ensemble.py +0 -2
test_ensemble.py
CHANGED
@@ -22,9 +22,7 @@ class TestEnsembleQAPipeline(Text2TextGenerationPipeline):
|
|
22 |
|
23 |
def postprocess(self, model_outputs):
|
24 |
results = {}
|
25 |
-
print(model_outputs)
|
26 |
for output in model_outputs:
|
27 |
-
print(output)
|
28 |
guess_text = output[0]
|
29 |
confidence = output[1]
|
30 |
results = {'guess': guess_text, 'confidence': confidence}
|
|
|
22 |
|
23 |
def postprocess(self, model_outputs):
|
24 |
results = {}
|
|
|
25 |
for output in model_outputs:
|
|
|
26 |
guess_text = output[0]
|
27 |
confidence = output[1]
|
28 |
results = {'guess': guess_text, 'confidence': confidence}
|