paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fixed documents not being saved after modification

Browse Source
This commit is contained in:
Jonas Winkler 2018-09-04 15:33:51 +02:00
parent c091eba26e
commit 30134034e2
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/documents/classifier.py
View File

@ -48,20 +48,26 @@ class DocumentClassifier(object):
def classify_document(self, document, classify_correspondent=False, classify_type=False, classify_tags=False): def classify_document(self, document, classify_correspondent=False, classify_type=False, classify_tags=False):
X = self.data_vectorizer.transform([preprocess_content(document.content)]) X = self.data_vectorizer.transform([preprocess_content(document.content)])
update_fields=()
if classify_correspondent: if classify_correspondent:
y_correspondent = self.correspondent_classifier.predict(X) y_correspondent = self.correspondent_classifier.predict(X)
correspondent = self.correspondent_binarizer.inverse_transform(y_correspondent)[0] correspondent = self.correspondent_binarizer.inverse_transform(y_correspondent)[0]
print("Detected correspondent:", correspondent) print("Detected correspondent:", correspondent)
document.correspondent = Correspondent.objects.filter(name=correspondent).first() document.correspondent = Correspondent.objects.filter(name=correspondent).first()
update_fields = update_fields + ("correspondent",)
if classify_type: if classify_type:
y_type = self.type_classifier.predict(X) y_type = self.type_classifier.predict(X)
type = self.type_binarizer.inverse_transform(y_type)[0] type = self.type_binarizer.inverse_transform(y_type)[0]
print("Detected document type:", type) print("Detected document type:", type)
document.type = DocumentType.objects.filter(name=type).first() document.document_type = DocumentType.objects.filter(name=type).first()
update_fields = update_fields + ("document_type",)
if classify_tags: if classify_tags:
y_tags = self.tags_classifier.predict(X) y_tags = self.tags_classifier.predict(X)
tags = self.tags_binarizer.inverse_transform(y_tags)[0] tags = self.tags_binarizer.inverse_transform(y_tags)[0]
print("Detected tags:", tags) print("Detected tags:", tags)
document.tags.add(*[Tag.objects.filter(name=t).first() for t in tags]) document.tags.add(*[Tag.objects.filter(name=t).first() for t in tags])
document.save(update_fields=update_fields)