Spaces:
Sleeping
Sleeping
clean classifyiers part 2
Browse files- classifiers/__init__.py +1 -0
- classifiers/llm.py +1 -1
- classifiers/tfidf.py +1 -1
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):
|