Update coreference_resolution.py
Browse files
coreference_resolution.py
CHANGED
|
@@ -228,7 +228,6 @@ class CoreferenceResolver:
|
|
| 228 |
'needs_llm_fallback': needs_llm,
|
| 229 |
'num_clusters': len(clusters),
|
| 230 |
'num_low_confidence': len(low_confidence_spans),
|
| 231 |
-
'preds': preds,
|
| 232 |
}
|
| 233 |
|
| 234 |
def _get_confidence_reason(self, avg_logit: float, min_logit: float, validation: Dict) -> str:
|
|
|
|
| 228 |
'needs_llm_fallback': needs_llm,
|
| 229 |
'num_clusters': len(clusters),
|
| 230 |
'num_low_confidence': len(low_confidence_spans),
|
|
|
|
| 231 |
}
|
| 232 |
|
| 233 |
def _get_confidence_reason(self, avg_logit: float, min_logit: float, validation: Dict) -> str:
|