TitleGenerator

This module provides functionality to generate titles.

Classes

class chemdatawriter.title_generator.TitleGenerator(title_generator_hf='tennessejoyce/titlewave-t5-base')

Generates titles using a specified model.

__init__(title_generator_hf='tennessejoyce/titlewave-t5-base')

Initialize the TitleGenerator with the given model.

Parameters:

title_generator_hf (str) – Model name or path to be used for title generation (default is ‘tennessejoyce/titlewave-t5-base’).

generate(text, max_length=30, num_beams=5)

Generate a title for the provided text.

Parameters:

• text (str) – The text for which a title needs to be generated.

• max_length (int) – The maximum length of the generated title (default is 30).

• num_beams (int) – The number of beams to be used during generation (default is 5).

Returns:

The generated title for the provided text.

Return type:

str