rupo package

Subpackages

Submodules

rupo.api module

class rupo.api.Engine(language='ru')[source]

Bases: object

classify_metre(text: str, language: str = 'ru') → str[source]
Parameters

  • text – текст.

  • language – язык.

Returns

его метр.

static count_syllables(word: str) → int[source]
Parameters

word – слово.

Returns

количество слогов в нём.

generate_markups(input_path: str, input_type: FileType, output_path: str, output_type: FileType) → None[source]

Генерация разметок по текстам.

Parameters

  • input_path – путь к папке/файлу с текстом.

  • input_type – тип файлов с текстов.

  • output_path – путь к файлу с итоговыми разметками.

  • output_type – тип итогового файла.

generate_poem(model_path: str, token_vocab_path: Optional[str] = None, stress_vocab_path: Optional[str] = None, metre_schema: str = '-+', rhyme_pattern: str = 'abab', n_syllables: int = 8, sampling_k: Optional[int] = None, beam_width: Optional[int] = None, seed: int = 1337, temperature: float = 1.0, last_text: str = '') → str[source]

Сгенерировать стих. Нужно задать либо sampling_k, либо beam_width.

Parameters

  • model_path – путь к модели.

  • token_vocab_path – путь к словарю.

  • stress_vocab_path – путь к словарю ударений.

  • metre_schema – схема метра.

  • rhyme_pattern – схема рифм.

  • n_syllables – количество слогов в строке.

  • sampling_k – top-k при семплинге

  • beam_width – ширина лучевого поиска.

  • seed – seed

  • temperature – температура генерации

  • last_text – последняя строчка

Returns

стих. None, если генерация не была успешной.

get_generator(model_path: str, token_vocab_path: str, stress_vocab_dump_path: str) → Generator[source]
get_improved_markup(text: str, language: str = 'ru') → Tuple[Markup, ClassificationResult][source]
Parameters

  • text – текст.

  • language – язык.

Returns

его разметка по словарю, классификатору метру и ML классификатору.

get_markup(text: str, language: str = 'ru') → Markup[source]
Parameters

  • text – текст.

  • language – язык.

Returns

его разметка по словарю.

get_stress_predictor(language='ru', stress_model_path: Optional[str] = None, raw_stress_dict_path=None, stress_trie_path=None, zalyzniak_dict='/home/docs/checkouts/readthedocs.org/user_builds/rupo/checkouts/latest/rupo/data/dict/zaliznyak.txt', cmu_dict='/home/docs/checkouts/readthedocs.org/user_builds/rupo/checkouts/latest/rupo/data/dict/cmu.txt')[source]
get_stresses(word: str, language: str = 'ru') → List[int][source]
Parameters

  • word – слово.

  • language – язык.

Returns

ударения слова.

get_vocabulary(dump_path: str, markup_path: str) → StressVocabulary[source]
get_word_rhymes(word: str, vocab_dump_path: str, markup_path: Optional[str] = None) → List[str][source]

Поиск рифмы для данного слова.

Parameters

  • word – слово.

  • vocab_dump_path – путь, куда сохраняется словарь.

  • markup_path – путь к разметкам.

Returns

список рифм.

static get_word_syllables(word: str) → List[str][source]
Parameters

word – слово.

Returns

его слоги.

is_rhyme(word1: str, word2: str) → bool[source]
Parameters

  • word1 – первое слово.

  • word2 – второе слово.

Returns

рифмуются ли слова.

load(stress_model_path: str, zalyzniak_dict: str, raw_stress_dict_path=None, stress_trie_path=None)[source]

rupo.settings module

rupo.test_api module

class rupo.test_api.TestApi(methodName='runTest')[source]

Bases: TestCase

classmethod setUpClass()[source]

Hook method for setting up class fixture before running tests in the class.

classmethod tearDownClass()[source]

Hook method for deconstructing the class fixture after running all tests in the class.

test_classify_metre()[source]
test_count_syllables()[source]
test_generate_poem()[source]
test_get_improved_markup()[source]
test_get_markup()[source]
test_get_word_rhymes()[source]
test_get_word_syllables()[source]
test_is_rhyme()[source]
test_stress()[source]

Module contents