Spaces:
Build error
Build error
Update competition/llama.py
Browse files- competition/llama.py +1 -7
competition/llama.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
# %%
|
2 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
|
3 |
from datasets import Dataset
|
4 |
import pandas as pd
|
@@ -137,9 +136,4 @@ tokenizer.save_pretrained('trained_llama_model')
|
|
137 |
|
138 |
# Evaluate the model with progress bar
|
139 |
eval_results = trainer.evaluate()
|
140 |
-
print(eval_results)
|
141 |
-
|
142 |
-
# %%
|
143 |
-
|
144 |
-
|
145 |
-
|
|
|
|
|
1 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
|
2 |
from datasets import Dataset
|
3 |
import pandas as pd
|
|
|
136 |
|
137 |
# Evaluate the model with progress bar
|
138 |
eval_results = trainer.evaluate()
|
139 |
+
print(eval_results)
|
|
|
|
|
|
|
|
|
|