Unverified Commit 00abc99a authored by Stella Biderman's avatar Stella Biderman Committed by GitHub
Browse files

Merge pull request #51 from anishthite/master

add Wikitext2 + wikitext103 data
parents b0db32bc 262fe250
import numpy as np
from scipy.stats import pearsonr, spearmanr
from sklearn.metrics import f1_score, matthews_corrcoef
from tqdm import auto as tqdm_lib
from . common import NLP_TASK, simple_accuracy_metric, yesno
class WikiText103(NLP_TASK):
NLP_PATH = "wikitext"
NLP_NAME = "wikitext-103-raw-v1"
def fewshot_description(self):
return ""
def doc_to_text(self, doc, include_target=True):
return doc['text']
def evaluate(self, docs, lm, provide_description, num_fewshot):
pass
class WikiText2(NLP_TASK):
NLP_PATH = "wikitext"
NLP_NAME = "wikitext-2-raw-v1"
def fewshot_description(self):
return ""
def doc_to_text(self, doc, include_target=True):
return doc['text']
def evaluate(self, docs, lm, provide_description, num_fewshot):
pass
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment