birbnetes/classification-service
birbnetes/classification-service
Archived
4
0
Fork 0
Code Issues Releases Activity

Fixed boolean memes
Some checks reported errors
continuous-integration/drone/push Build was killed
Details

Browse Source
This commit is contained in:
Pünkösd Marcell 2020-05-29 17:50:18 +02:00
parent 88ddc35741
commit c3e394d0ca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
classification_service/mule.py
View File

@ -70,7 +70,7 @@ def run_classification(task, target_class_name: str):
results = { results = {
"tag": task['tag'], "tag": task['tag'],
"model": task['model'], "model": task['model'],
"is_target": class_id == target_id, "is_target": bool(class_id == target_id),
"probability": probability[target_id] "probability": probability[target_id]
} }