utils API

HookCloser

class HookCloser(model_wrapper)[source]

make_zip_downloader

make_zip_downloader(URL, file_list=None, resource_name=None)[source]

This function is used to make a zipfile downloader for data.

Parameters

URL (str) –

visualizer

visualizer(idx, x_orig, y_orig, x_adv, y_adv, info, stream_writer, tokenizer, key_len=25, val_len=10)[source]

Visualization tools used in DefaultAttackEval.

result_visualizer

result_visualizer(result, stream_writer)[source]

Visualization tools used in DefaultAttackEval.

update_label

update_label(dataset, labels_to_labels)[source]
Parameters
  • dataset (datasets) – The huggingface datasets you use.

  • labels_to_labels (dict) – map the origin labels to the labels you want.

Package Requirements
  • datasets

get_language

get_language(obj_list)[source]

check_language

check_language(obj_list, lang_tag)[source]

language_by_name

language_by_name(name)[source]