taratrankennedy
commited on
Commit
•
3f325ac
1
Parent(s):
78247ea
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"
|