Spaces:
Runtime error
Runtime error
Merge pull request #44 from huggingface/add-question-answering
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ TASK_TO_ID = {
|
|
| 37 |
"binary_classification": 1,
|
| 38 |
"multi_class_classification": 2,
|
| 39 |
"entity_extraction": 4,
|
| 40 |
-
|
| 41 |
"translation": 6,
|
| 42 |
"summarization": 8,
|
| 43 |
}
|
|
@@ -51,7 +51,7 @@ TASK_TO_DEFAULT_METRICS = {
|
|
| 51 |
"accuracy",
|
| 52 |
],
|
| 53 |
"entity_extraction": ["precision", "recall", "f1", "accuracy"],
|
| 54 |
-
"extractive_question_answering": [],
|
| 55 |
"translation": ["sacrebleu"],
|
| 56 |
"summarization": ["rouge1", "rouge2", "rougeL", "rougeLsum"],
|
| 57 |
"image_binary_classification": ["f1", "precision", "recall", "auc", "accuracy"],
|
|
@@ -71,7 +71,7 @@ AUTOTRAIN_TASK_TO_LANG = {
|
|
| 71 |
|
| 72 |
|
| 73 |
SUPPORTED_TASKS = list(TASK_TO_ID.keys())
|
| 74 |
-
UNSUPPORTED_TASKS = [
|
| 75 |
|
| 76 |
# Extracted from utils.get_supported_metrics
|
| 77 |
# Hardcoded for now due to speed / caching constraints
|
|
|
|
| 37 |
"binary_classification": 1,
|
| 38 |
"multi_class_classification": 2,
|
| 39 |
"entity_extraction": 4,
|
| 40 |
+
"extractive_question_answering": 5,
|
| 41 |
"translation": 6,
|
| 42 |
"summarization": 8,
|
| 43 |
}
|
|
|
|
| 51 |
"accuracy",
|
| 52 |
],
|
| 53 |
"entity_extraction": ["precision", "recall", "f1", "accuracy"],
|
| 54 |
+
"extractive_question_answering": ["f1", "exact_match"],
|
| 55 |
"translation": ["sacrebleu"],
|
| 56 |
"summarization": ["rouge1", "rouge2", "rougeL", "rougeLsum"],
|
| 57 |
"image_binary_classification": ["f1", "precision", "recall", "auc", "accuracy"],
|
|
|
|
| 71 |
|
| 72 |
|
| 73 |
SUPPORTED_TASKS = list(TASK_TO_ID.keys())
|
| 74 |
+
UNSUPPORTED_TASKS = []
|
| 75 |
|
| 76 |
# Extracted from utils.get_supported_metrics
|
| 77 |
# Hardcoded for now due to speed / caching constraints
|