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 |
)
|