Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -274,19 +274,24 @@ class EventAnalyzer:
|
|
274 |
))
|
275 |
|
276 |
# Adjust confidence based on entity frequency
|
277 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
278 |
SELECT AVG(frequency) as avg_freq
|
279 |
FROM entities
|
280 |
WHERE entity_text IN (
|
281 |
SELECT DISTINCT entity_text
|
282 |
FROM entities
|
283 |
-
WHERE entity_text IN ({})
|
284 |
)
|
285 |
-
|
286 |
-
|
287 |
-
|
288 |
-
*entities["locations"]
|
289 |
-
]))))
|
290 |
|
291 |
avg_frequency = cursor.fetchone()[0] or 1
|
292 |
frequency_boost = min(0.2, (avg_frequency - 1) * 0.05) # Max 0.2 boost for frequency
|
|
|
274 |
))
|
275 |
|
276 |
# Adjust confidence based on entity frequency
|
277 |
+
entity_params = [
|
278 |
+
*entities["people"],
|
279 |
+
*entities["organizations"],
|
280 |
+
*entities["locations"]
|
281 |
+
]
|
282 |
+
|
283 |
+
cursor = self.relationship_engine.conn.execute(
|
284 |
+
f'''
|
285 |
SELECT AVG(frequency) as avg_freq
|
286 |
FROM entities
|
287 |
WHERE entity_text IN (
|
288 |
SELECT DISTINCT entity_text
|
289 |
FROM entities
|
290 |
+
WHERE entity_text IN ({','.join(['?']*len(entity_params))})
|
291 |
)
|
292 |
+
''',
|
293 |
+
entity_params # Pass parameters here
|
294 |
+
)
|
|
|
|
|
295 |
|
296 |
avg_frequency = cursor.fetchone()[0] or 1
|
297 |
frequency_boost = min(0.2, (avg_frequency - 1) * 0.05) # Max 0.2 boost for frequency
|