Source code for malaya.nsfw

from malaya.function import check_file
from malaya.supervised import classification
from malaya.text.lexicon import nsfw
from malaya.path import PATH_NSFW, S3_PATH_NSFW
import json

label = ['sex', 'gambling', 'negative']


def lexicon(**kwargs):
    """
    Load Lexicon NSFW model.

    Returns
    -------
    result : malaya.text.lexicon.nsfw.Lexicon class
    """

    path = check_file(PATH_NSFW['lexicon'], S3_PATH_NSFW['lexicon'], **kwargs)
    with open(path['model']) as fopen:
        corpus = json.load(fopen)
    return nsfw.Lexicon(corpus)


def multinomial(**kwargs):
    """
    Load multinomial NSFW model.

    Returns
    -------
    result : malaya.model.ml.BAYES class
    """
    return classification.multinomial(
        path=PATH_NSFW,
        s3_path=S3_PATH_NSFW,
        module='nsfw',
        label=label,
        **kwargs
    )