mirror of https://github.com/doccano/doccano.git
pythonannotation-tooldatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learning
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
454 B
18 lines
454 B
from model_mommy import mommy
|
|
|
|
from projects.models import (
|
|
DOCUMENT_CLASSIFICATION,
|
|
SEQ2SEQ,
|
|
SEQUENCE_LABELING,
|
|
SPEECH2TEXT,
|
|
)
|
|
|
|
|
|
def make_annotation(task, doc, user, **kwargs):
|
|
annotation_model = {
|
|
DOCUMENT_CLASSIFICATION: "Category",
|
|
SEQUENCE_LABELING: "Span",
|
|
SEQ2SEQ: "TextLabel",
|
|
SPEECH2TEXT: "TextLabel",
|
|
}.get(task)
|
|
return mommy.make(annotation_model, example=doc, user=user, **kwargs)
|