rl4lms.envs.text_generation.summ_metrics.summa_c module
Code taken from https://github.com/tingofurro/summac
-
rl4lms.envs.text_generation.summ_metrics.summa_c.batcher(iterator, batch_size=4, progress=False)[source]
-
rl4lms.envs.text_generation.summ_metrics.summa_c.card_to_name(card)[source]
-
rl4lms.envs.text_generation.summ_metrics.summa_c.name_to_card(name)[source]
-
rl4lms.envs.text_generation.summ_metrics.summa_c.get_neutral_idx(ent_idx, con_idx)[source]
-
class rl4lms.envs.text_generation.summ_metrics.summa_c.SummaCImager(model_name='mnli', granularity='paragraph', use_cache=True, max_doc_sents=100, **kwargs)[source]
Bases: object
-
__init__(model_name='mnli', granularity='paragraph', use_cache=True, max_doc_sents=100, **kwargs)[source]
-
load_nli()[source]
-
split_sentences(text)[source]
-
split_2sents(text)[source]
-
split_paragraphs(text)[source]
-
split_text(text, granularity='sentence')[source]
-
build_image(original, generated)[source]
-
get_cache_file()[source]
-
save_cache()[source]
-
load_cache()[source]
-
class rl4lms.envs.text_generation.summ_metrics.summa_c.SummaCZS(model_name='mnli', granularity='paragraph', op1='max', op2='mean', use_ent=True, use_con=True, imager_load_cache=True, **kwargs)[source]
Bases: object
-
__init__(model_name='mnli', granularity='paragraph', op1='max', op2='mean', use_ent=True, use_con=True, imager_load_cache=True, **kwargs)[source]
-
save_imager_cache()[source]
-
score_one(original, generated)[source]
-
score(sources, generateds, **kwargs)[source]