mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-05-01 11:19:32 -05:00
21 lines
516 B
Python
21 lines
516 B
Python
from django.core.management.base import BaseCommand
|
|
|
|
from documents.tasks import train_classifier
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
help = """
|
|
Trains the classifier on your data and saves the resulting models to a
|
|
file. The document consumer will then automatically use this new model.
|
|
""".replace(
|
|
" ",
|
|
"",
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
BaseCommand.__init__(self, *args, **kwargs)
|
|
|
|
def handle(self, *args, **options):
|
|
train_classifier()
|