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):
|