Update model_evaluation.py
Browse files- model_evaluation.py +1 -1
model_evaluation.py
CHANGED
|
@@ -4,7 +4,7 @@ from transformers import pipeline, GPT2Tokenizer
|
|
| 4 |
from sentence_transformers import SentenceTransformer, util
|
| 5 |
|
| 6 |
# Define paths and models
|
| 7 |
-
filename = "
|
| 8 |
retrieval_model_name = 'output/sentence-transformer-finetuned/' #using a prefine-tuned model
|
| 9 |
gpt2_model_name = "gpt2"
|
| 10 |
csv_file_path = "train_dataset.csv"
|
|
|
|
| 4 |
from sentence_transformers import SentenceTransformer, util
|
| 5 |
|
| 6 |
# Define paths and models
|
| 7 |
+
filename = "output_chess_details.txt"
|
| 8 |
retrieval_model_name = 'output/sentence-transformer-finetuned/' #using a prefine-tuned model
|
| 9 |
gpt2_model_name = "gpt2"
|
| 10 |
csv_file_path = "train_dataset.csv"
|