Mehrdad-S commited on
Commit
2255cd8
·
verified ·
1 Parent(s): 6369f69

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[:200]")
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[:2]")
8
 
9
  scores = []
10
  for row in dataset: