Update scoring/specificity.py
Browse files- scoring/specificity.py +2 -2
scoring/specificity.py
CHANGED
|
@@ -20,7 +20,7 @@ class PredictionsResponse(BaseModel):
|
|
| 20 |
results: List[Dict[str, Union[str, float]]]
|
| 21 |
|
| 22 |
class BatchPredictionScoreItem(BaseModel):
|
| 23 |
-
|
| 24 |
score: float
|
| 25 |
|
| 26 |
# Model environment variables
|
|
@@ -114,7 +114,7 @@ def predict_batch(items: ProblematicList):
|
|
| 114 |
|
| 115 |
results.append(
|
| 116 |
BatchPredictionScoreItem(
|
| 117 |
-
|
| 118 |
score=score_class_1
|
| 119 |
)
|
| 120 |
)
|
|
|
|
| 20 |
results: List[Dict[str, Union[str, float]]]
|
| 21 |
|
| 22 |
class BatchPredictionScoreItem(BaseModel):
|
| 23 |
+
problematic: str
|
| 24 |
score: float
|
| 25 |
|
| 26 |
# Model environment variables
|
|
|
|
| 114 |
|
| 115 |
results.append(
|
| 116 |
BatchPredictionScoreItem(
|
| 117 |
+
problematic=batch_texts[j],
|
| 118 |
score=score_class_1
|
| 119 |
)
|
| 120 |
)
|