Terry Zhang
commited on
Commit
·
f5aa578
1
Parent(s):
8589546
update model path
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -31,7 +31,7 @@ def baseline_model(dataset_length: int):
|
|
| 31 |
def tree_classifier(test_dataset: dict, model: str):
|
| 32 |
texts = test_dataset["quote"]
|
| 33 |
|
| 34 |
-
model_path = f"text_models/{model}.skops"
|
| 35 |
|
| 36 |
model = load(model_path,
|
| 37 |
trusted=[
|
|
|
|
| 31 |
def tree_classifier(test_dataset: dict, model: str):
|
| 32 |
texts = test_dataset["quote"]
|
| 33 |
|
| 34 |
+
model_path = f"tasks/text_models/{model}.skops"
|
| 35 |
|
| 36 |
model = load(model_path,
|
| 37 |
trusted=[
|