simondh commited on
Commit
720c911
·
1 Parent(s): 0f1938f

clean classifyiers part 2

Browse files
classifiers/__init__.py CHANGED
@@ -1,3 +1,4 @@
1
  from .tfidf import TFIDFClassifier
 
2
 
3
  __all__ = ['TFIDFClassifier', 'LLMClassifier']
 
1
  from .tfidf import TFIDFClassifier
2
+ from .llm import LLMClassifier
3
 
4
  __all__ = ['TFIDFClassifier', 'LLMClassifier']
classifiers/llm.py CHANGED
@@ -10,7 +10,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
10
  from typing import List, Dict, Any, Optional
11
  from prompts import CATEGORY_SUGGESTION_PROMPT, TEXT_CLASSIFICATION_PROMPT
12
 
13
- from base import BaseClassifier
14
 
15
 
16
 
 
10
  from typing import List, Dict, Any, Optional
11
  from prompts import CATEGORY_SUGGESTION_PROMPT, TEXT_CLASSIFICATION_PROMPT
12
 
13
+ from .base import BaseClassifier
14
 
15
 
16
 
classifiers/tfidf.py CHANGED
@@ -10,7 +10,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
10
  from typing import List, Dict, Any, Optional
11
  from prompts import CATEGORY_SUGGESTION_PROMPT, TEXT_CLASSIFICATION_PROMPT
12
 
13
- from base import BaseClassifier
14
 
15
 
16
  class TFIDFClassifier(BaseClassifier):
 
10
  from typing import List, Dict, Any, Optional
11
  from prompts import CATEGORY_SUGGESTION_PROMPT, TEXT_CLASSIFICATION_PROMPT
12
 
13
+ from .base import BaseClassifier
14
 
15
 
16
  class TFIDFClassifier(BaseClassifier):