def classify(X): global classifier model_name = 'TC_GENERAL_V131' model_path = ModelFetcher.get_model_path(model_name) if not classifier: if not os.path.exists(model_path): ModelFetcher.download(model_name) classifier = TextClassifier.load(model_path) sentence = Sentence(X) classifier.predict(sentence) labels = sentence.labels return labels
def classify(X): global classifier if not classifier: if os.path.exists(model_path): classifier = TextClassifier.load(model_path) else: logger.error( f"Could not load model at {model_path}.\n" f"Download model with \"underthesea download {UTSModel.tc_general.value}\".") sys.exit(1) sentence = Sentence(X) classifier.predict(sentence) labels = sentence.labels return labels
def sentiment(X): global classifier model_name = 'SA_BANK_V131' model_path = ModelFetcher.get_model_path(model_name) if not classifier: if not os.path.exists(model_path): ModelFetcher.download(model_name) classifier = TextClassifier.load(model_path) sentence = Sentence(X) classifier.predict(sentence) labels = sentence.labels if not labels: return None labels = [label.value for label in labels] return labels
def sentiment(text): global classifier if not classifier: if os.path.exists(model_path): classifier = TextClassifier.load(model_path) else: logger.error( f"Could not load model at {model_path}.\n" f"Download model with \"underthesea download {UTSModel.sa_bank.value}\"." ) sys.exit(1) sentence = Sentence(text) classifier.predict(sentence) labels = sentence.labels if labels is None: return None return [label.value for label in labels]
def sentiment(text): global classifier if not classifier: if os.path.exists(model_path): classifier = TextClassifier.load(model_path) else: logger.error( f"Could not load model at {model_path}.\n" f"Download model with \"underthesea download {UTSModel.sa_general.value}\"." ) sys.exit(1) sentence = Sentence(text) classifier.predict(sentence) label = sentence.labels[0] if label == "1": label = "negative" if label == "0": label = "positive" return label
def sentiment(X): global classifier model_name = 'SA_GENERAL_V131' model_path = ModelFetcher.get_model_path(model_name) if not classifier: if not os.path.exists(model_path): ModelFetcher.download(model_name) classifier = TextClassifier.load(model_path) sentence = Sentence(X) classifier.predict(sentence) labels = sentence.labels try: label_map = {'POS': 'positive', 'NEG': 'negative'} label = label_map[labels[0]] return label except Exception: return None
from underthesea.corpus.data import Sentence from underthesea.models.text_classifier import TextClassifier model_folder = "tmp/classification_svm_vntc" print(f"Load model from {model_folder}") classifer = TextClassifier.load(model_folder) print(f"Model is loaded.") def predict(text): print(f"\nText: {text}") sentence = Sentence(text) classifer.predict(sentence) labels = sentence.labels print(f"Labels: {labels}") predict('Huawei có thể không cần Google, nhưng sẽ ra sao nếu thiếu ARM ?') predict( 'Trưởng phòng GD&ĐT xin lỗi vụ học sinh nhận khen thưởng là tờ giấy A4')