paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-07-28 18:24:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

don't load sklearn libraries unless needed

Browse Source
This commit is contained in:
jonaswinkler
2021-02-04 15:15:11 +01:00
parent ef29f5c512
commit d08a530701
4 changed files with 23 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
src/documents/tests/data/model.pickle Normal file
View File

Binary file not shown.

9
src/documents/tests/test_classifier.py
View File

@@ -130,6 +130,15 @@ class TestClassifier(DirectoriesMixin, TestCase):
new_classifier.reload()
self.assertFalse(new_classifier.train())
@override_settings(MODEL_FILE=os.path.join(os.path.dirname(__file__), "data", "model.pickle"))
def test_load_and_classify(self):
self.generate_test_data()
new_classifier = DocumentClassifier()
new_classifier.reload()
self.assertCountEqual(new_classifier.predict_tags(self.doc2.content), [45, 12])
def test_one_correspondent_predict(self):
c1 = Correspondent.objects.create(name="c1", matching_algorithm=Correspondent.MATCH_AUTO)
doc1 = Document.objects.create(title="doc1", content="this is a document from c1", correspondent=c1, checksum="A")