Spaces:

iSemantics
/

ner-demo-evaluate

Runtime error

elshehawy commited on Mar 2

Commit

bd639d6

•

1 Parent(s): d3f1526

use subset of 16 examples only

Files changed (1) hide show

evaluate_data.py CHANGED Viewed

@@ -22,7 +22,7 @@ tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 data_path = './data/merged_dataset/'
 test = load_from_disk(data_path)['test']
-test = Dataset.from_dict(test[:105])
 feature_path = './data/ner_feature.pickle'

 data_path = './data/merged_dataset/'
 test = load_from_disk(data_path)['test']
+test = Dataset.from_dict(test[:16])
 feature_path = './data/ner_feature.pickle'