Terry Zhang commited on
Commit
85ec854
·
1 Parent(s): f5aa578

change textprocessor import

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -1
tasks/text.py CHANGED
@@ -4,7 +4,6 @@ from datasets import load_dataset
4
  from sklearn.metrics import accuracy_score
5
  import random
6
  from skops.io import load
7
- from .utils.text_preprocessor import TextPreprocessor
8
 
9
 
10
  from .utils.evaluation import TextEvaluationRequest
@@ -29,6 +28,8 @@ def baseline_model(dataset_length: int):
29
  return predictions
30
 
31
  def tree_classifier(test_dataset: dict, model: str):
 
 
32
  texts = test_dataset["quote"]
33
 
34
  model_path = f"tasks/text_models/{model}.skops"
 
4
  from sklearn.metrics import accuracy_score
5
  import random
6
  from skops.io import load
 
7
 
8
 
9
  from .utils.evaluation import TextEvaluationRequest
 
28
  return predictions
29
 
30
  def tree_classifier(test_dataset: dict, model: str):
31
+ from .utils.text_preprocessor import TextPreprocessor
32
+
33
  texts = test_dataset["quote"]
34
 
35
  model_path = f"tasks/text_models/{model}.skops"