Update tasks/text.py

tasks/text.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter
 from datetime import datetime
 from datasets import load_dataset
 from sklearn.feature_extraction.text import TfidfVectorizer
-from sklearn.naive_bayes import MultinomialNB
+from sklearn.svm import SVC
 from sklearn.metrics import accuracy_score
 
 from .utils.evaluation import TextEvaluationRequest
@@ -10,18 +10,18 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 router = APIRouter()
 
-DESCRIPTION = "Naive Bayes Text Classifier"
-ROUTE = "/text"
+DESCRIPTION = "SVM Text Classifier with TF-IDF"
+ROUTE = "/text_svm"
 
 @router.post(ROUTE, tags=["Text Task"], 
              description=DESCRIPTION)
-async def evaluate_text(request: TextEvaluationRequest):
+async def evaluate_text_svm(request: TextEvaluationRequest):
     """
     Evaluate text classification for climate disinformation detection.
     
-    Current Model: Naive Bayes Classifier
+    Current Model: SVM Classifier
     - Uses TF-IDF for text vectorization
-    - Trains and evaluates a Multinomial Naive Bayes model
+    - Trains and evaluates a Support Vector Machine (SVM) model
     """
     # Get space info
     username, space_url = get_space_info()
@@ -60,8 +60,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     train_vectors = vectorizer.fit_transform(train_texts)
     test_vectors = vectorizer.transform(test_texts)
 
-    # Train Naive Bayes Classifier
-    model = MultinomialNB()
+    # Train SVM Classifier
+    model = SVC(kernel="linear", probability=True)
     model.fit(train_vectors, train_labels)
 
     # Start tracking emissions