up / prompt_generate.py
Pengfa Li
Update prompt_generate.py
674554d verified
raw
history blame contribute delete
694 Bytes
import random
import json
from difflib import SequenceMatcher
import pandas as pd
def generate_prompt_with_examples(text, label, n, start_index=500, end_index=1000):
text_len = len(text)
label_len = len(label)
end_index = min(end_index, text_len, label_len)
if start_index >= end_index:
return "none"
available_examples = end_index - start_index
n = min(n, available_examples)
prompt = ""
random_indices = random.sample(range(start_index, end_index), n)
for i in random_indices:
text_prompt = text.iloc[i]
triple_prompt = label.iloc[i]
prompt += text_prompt + '\n' + str(triple_prompt) + '\n'
return prompt