promptbench icon indicating copy to clipboard operation
promptbench copied to clipboard

examples basic.ipyn

Open zl-comment opened this issue 1 year ago • 0 comments

from tqdm import tqdm for prompt in prompts: preds = [] labels = [] for data in tqdm(dataset): # process input input_text = pb.InputProcess.basic_format(prompt, data) label = data['label'] print(type(input_text)) raw_pred = model(input_text)

    # process output
    pred = pb.OutputProcess.cls(raw_pred, proj_func)
    preds.append(pred)
    labels.append(label)

# evaluate
score = pb.Eval.compute_cls_accuracy(preds, labels)
print(f"{score:.3f}, {prompt}")

but have some quenstion about TypeError Traceback (most recent call last) Cell In[22], line 10 8 label = data['label'] 9 print(type(input_text)) ---> 10 raw_pred = model(input_text) 12 # process output 13 pred = pb.OutputProcess.cls(raw_pred, proj_func) end in

zl-comment avatar Apr 26 '24 06:04 zl-comment