mirror of https://github.com/doccano/doccano.git
pythondatasetnatural-language-processingdata-labelingmachine-learningannotation-tooldatasetsactive-learningtext-annotation
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.
34 lines
1.2 KiB
34 lines
1.2 KiB
from auto_labeling_pipeline.models import RequestModelFactory
|
|
from django.core.exceptions import ValidationError
|
|
from django.db import models
|
|
|
|
from api.models import Project
|
|
|
|
|
|
class AutoLabelingConfig(models.Model):
|
|
model_name = models.CharField(max_length=100)
|
|
model_attrs = models.JSONField(default=dict)
|
|
template = models.TextField(default='')
|
|
label_mapping = models.JSONField(default=dict, blank=True)
|
|
project = models.ForeignKey(
|
|
to=Project,
|
|
on_delete=models.CASCADE,
|
|
related_name='auto_labeling_config'
|
|
)
|
|
default = models.BooleanField(default=False)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
def __str__(self):
|
|
return self.model_name
|
|
|
|
def clean_fields(self, exclude=None):
|
|
super().clean_fields(exclude=exclude)
|
|
try:
|
|
RequestModelFactory.find(self.model_name)
|
|
except NameError:
|
|
message = f'The specified model name {self.model_name} does not exist.'
|
|
raise ValidationError(message)
|
|
except Exception:
|
|
message = 'The attributes does not match the model.'
|
|
raise ValidationError(message)
|