Source code for underthesea.pipeline.classification
# -*- coding: utf-8 -*-
[docs]def classify(X, domain=None):
"""
Text classification
Parameters
==========
X: {unicode, str}
raw sentence
domain: {None, 'bank'}
domain of text
* None: general domain
* bank: bank domain
Returns
=======
tokens: list
categories of sentence
"""
if X == "":
return None
if domain == 'bank':
from underthesea.pipeline.classification import bank
return bank.classify(X)
from underthesea.pipeline.classification import vntc
return vntc.classify(X)