Mehrdad-S commited on
Commit
aece1af
·
verified ·
1 Parent(s): be77ede

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +1 -1
evaluate.py CHANGED
@@ -4,7 +4,7 @@ from sentence_transformers import SentenceTransformer, util
4
  def evaluate_model(model_name):
5
  try:
6
  model = SentenceTransformer(model_name)
7
- dataset = load_dataset("arshiaafshani/persian-natural-fluently", split="train[:2]")
8
 
9
  scores = []
10
  for row in dataset:
 
4
  def evaluate_model(model_name):
5
  try:
6
  model = SentenceTransformer(model_name)
7
+ dataset = load_dataset("arshiaafshani/persian-natural-fluently", split="train[:10]")
8
 
9
  scores = []
10
  for row in dataset: