o 6tih@s`ddlZddlZGdddZddZdededed dfd d Zdd d Zedkr.edSdS)Nc@seZdZddZdS) FunctionTagcCs ||_dS)N)value)selfrr[/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/afrobench/afrisenti/utils.py__init__s zFunctionTag.__init__N)__name__ __module__ __qualname__rrrrrrs rcCs0dd|dd|ddd|dd }||S) NzZDoes this statement; {{tweet}} have a Neutral, Positive or Negative sentiment? Labels onlyz Does this zS statement; '{{tweet}}' have a Neutral, Positive or Negative sentiment? Labels onlyzYou are an assistant able to detect sentiments in tweets. Given the sentiment labels Neutral, Positive or Negative; what is the sentiment of the zC statement below? Return only the labels. text: {{tweet}} label:zLabel the following text as Neutral, Positive, or Negative. Provide only the label as your response. text: {{tweet}} label: zIYou are tasked with performing sentiment classification on the following aZ text. For each input, classify the sentiment as positive, negative, or neutral. Use the following guidelines: Positive: The text expresses happiness, satisfaction, or optimism. Negative: The text conveys disappointment, dissatisfaction, or pessimism. Neutral: The text is factual, objective, or without strong emotional undertones. If the text contains both positive and negative sentiments, choose the dominant sentiment. For ambiguous or unclear sentiments, select the label that best reflects the overall tone. Please provide a single classification for each input. text: {{tweet}} label: prompt_1prompt_2prompt_3prompt_4prompt_5r)modelang prompt_maprrr prompt_func s  r output_dir overwriterreturnc Cs4g}ddddddddd d d d d dd}|D]p}z`d|d}d|d|}d}t|dddkrB|||t|||d} n|||d} t|d|d||rVdnddd} | dtj| | ddWd n1sswYWqty| |Yqwt |d!krtd"d# |d S)$z Generate a yaml file for each language. :param output_dir: The directory to output the files to. :param overwrite: Whether to overwrite files if they already exist. AmhariczAlgerian ArabiczMoroccan ArabicHausaIgbo KinyarwandaOromozNigerian PidginzMozambique PortugueseSwahiliTigrinyaXithongaTwiYoruba)amharqaryhauibokinormpcmporswatirtsotwiyor afrisenti_z.yaml_ afrisenti)includetask dataset_name doc_to_text)r5r6r7/wxutf8)encodingz# Generated by utils.py T) allow_unicodeNrzJFiles were not created because they already exist (use --overwrite flag): z, ) keysintsplitropenwriteyamldumpFileExistsErrorappendlenjoin) rrrerr languagesr file_name task_name yaml_template yaml_detailsfrrrgen_lang_yamls#sn       rQcCs`t}|jddddd|jdddd |jd d gd d d|}t|j|j|jddS)z9Parse CLI args and generate language-specific yaml files.z --overwriteT store_truez%Overwrite files if they already exist)defaultactionhelpz --output-dirz./z Directory to write yaml files to)rSrUz--moder r z Prompt number)rSchoicesrU)rrrN)argparseArgumentParser add_argument parse_argsrQrrr)parserargsrrrmainbs(r]__main__)rN) rWrDrrstrboolrQr]rrrrrs ?