Update tasks/text.py

tasks/text.py

@@ -45,10 +45,13 @@ async def evaluate_text(request: TextEvaluationRequest):
     # Convert string labels to integers
     dataset = dataset.map(lambda x: {"label": LABEL_MAPPING[x["label"]]})
 
+
+    
     # Split dataset
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
-    
+    train_dataset = train_test["train"]
+    y_train=train_dataset['label']
     tfidf_vect = TfidfVectorizer(stop_words = 'english')
     
     tfidf_train = tfidf_vect.fit_transform(train_dataset['quote'])
@@ -68,7 +71,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     true_labels = test_dataset["label"]
     LR = LogisticRegression(class_weight='balanced', max_iter=20, random_state=1234,
                    solver='liblinear')
-    LR.fit(pd.DataFrame.sparse.from_spmatrix(tfidf_train), pd.DataFrame(y_train_v))
+    LR.fit(pd.DataFrame.sparse.from_spmatrix(tfidf_train), pd.DataFrame(y_train))
     predictions=LR.predict(pd.DataFrame.sparse.from_spmatrix(tfidf_test))
     
     #--------------------------------------------------------------------------------------------