mindmeld.augmentation module

This module contains the data augmentation processes for MindMeld.

exception mindmeld.augmentation.UnsupportedLanguageError

Bases: Exception

class mindmeld.augmentation.Augmentor(language, paths, path_suffix, resource_loader)

Bases: abc.ABC

Abstract Augmentor class.

augment(**kwargs)

Augments queries given initial queries in application.

augment_queries(queries)

Generates augmented data given application queries.

Parameters:queries (list) -- List of queries. Returns:List of augmented queries. Return type:augmented_queries (list)

class mindmeld.augmentation.AugmentorFactory(config, language, resource_loader)

Bases: object

Creates an Augmentor object.

config

dict -- A model configuration.

language

str -- Language for data augmentation.

resource_loader

object -- Resource Loader object for the application.

create_augmentor()

Creates an augmentor instance using the provided configuration

Returns:An Augmentor class Return type:Augmentor Raises:ValueError -- When model configuration is invalid or required key is missing

class mindmeld.augmentation.EnglishParaphraser(batch_size, language, retain_entities, paths, path_suffix, resource_loader)

Bases: mindmeld.augmentation.Augmentor

Paraphraser class for generating English paraphrases.

augment_queries(queries, **kwargs)

Generates augmented data given application queries.

Parameters:queries (list) -- List of queries. Returns:List of augmented queries. Return type:augmented_queries (list)

class mindmeld.augmentation.MultiLingualParaphraser(batch_size, language, retain_entities, paths, path_suffix, resource_loader)

Bases: mindmeld.augmentation.Augmentor

Paraphraser class for generating paraphrases based on language code of the app (currently supports: French, Italian, Portuguese, Romanian and Spanish).

augment_queries(processed_queries)

Generates augmented data given application queries.

Parameters:queries (list) -- List of queries. Returns:List of augmented queries. Return type:augmented_queries (list)

mindmeld.augmentation.register_all_augmentors()