metadata

language:
  - tw
tags:
  - t5
license: afl-3.0

Hotel review multi-aspect sentiment classification using T5

We fine tune a T5 pretrained model to generate multi-aspect sentiment classes. The outputs are whole sentiment, aspect, and aspect+sentiment.

T5情緒面向分類多任務，依據中文簡體孟子T5預訓練模型微調，訓練資料集只有3萬筆，做NLP研究與課程的範例模型用途。

如何測試

在右側測試區輸入不同的任務文字

    範例1:
    面向::早餐可以吃的饱，但是东西没了，不一定会补
    範例2:
    面向情绪::房间空调系统有烟味，可考虑做调整
    範例3:
    整体情绪::位置离逢甲很近

資料集: 資料集蒐集自線上訂房網站的顧客留言10050筆，整理成3項任務，總筆數變成為3倍，共有30150筆(資料由本實驗室成員YYChang蒐集)。

輸入與輸出格式:有三個種類任務分別為:

'整体情绪::' '面向::', '面向情绪::'

    舉例如下:
    整体情绪::因为防疫期间早餐要在房内用餐，但房内电视下的平台有点窄，有点不方便,负面情绪
    整体情绪::只是隔音有点不好,负面情绪
    整体情绪::订的是豪华家庭房，空间还算大,正面情绪
    整体情绪::床大,正面情绪
    
    面向::房间有奇怪的味道,"整洁舒适面向,设施面向"
    面向::干净、舒适、亲切，价钱好～,"整洁舒适面向,性价比面向"
    面向::位置便利，可以在附近悠闲散步，至市区也不远，又临近大海，住得十分舒服。,"整洁舒适面向,地点面向"
    
    面向情绪::反应无效,服务面向的负面情绪
    面向情绪::床其实还蛮好睡，枕头床被还算干净，至少不会让皮肤痒。离火车站市场闹区近。,"整洁舒适面向的正面情绪,设施面向的正面情绪,地点面向的正面情绪"
    面向情绪::设备真的太旧了，灯光太暗了。,设施面向的负面情绪
    面向情绪::住四天，没人打扫清洁，第一天有盥洗用品，其余就没补充，热水供应不正常，交通尚可。,"整洁舒适面向的负面情绪,设施面向的负面情绪,地点面向的正面情绪"
    面向情绪::饭店太过老旧，房内桌子衣橱近乎溃烂，浴室有用过未清的毛巾，排水孔有近半垃圾未清，马桶肮脏，未提供浴巾，莲蓬头只能手持无法挂著墙上使用，空调无法控制，壁纸剥落，走道昏暗，近车站。,"整洁舒适面向的负面情绪,设施面向的负面情绪,地点面向的正面情绪"

預訓練模型: 目前初步先使用"Langboat/mengzi-t5-base"簡體中文預訓練模型加以微調。由"Langboat/mengzi-t5-base"官網資訊得知是由簡體中文語料所訓練，因此我們將繁體中文留言先轉成簡體中文，再進行微調訓練。

訓練平台: 使用Google colab Tesla T4 GPU進行了3 epochs訓練，費時55分鐘，val_loss約為0.0315，初步實驗，仍有很大的改善空間。

未來改善工作:下一階段會進行數據增強(由於蒐集的語料是不平衡),以及使用Google的mt5繁體簡體中文預訓練模型加以微調，微調語料就可直接使用繁體中文。

使用範例:(輸入繁體中文需先將文字轉為簡體中文，再丟給模型產出輸出文字)

載入模型(使用的是simplet5套件)

#pip install simplet5
from simplet5 import SimpleT5
model = SimpleT5()
model.load_model("t5","clhuang/t5-hotel-review-sentiment", use_gpu=False)

整體情緒分類任務

text="整体情绪::位置离逢甲很近"
model.predict(text)
#['正面情绪']

面向分類任務

text="面向::早餐可以吃的饱，但是东西没了，不一定会补"
model.predict(text)
#['服务面向']

面向分類+情绪分類任務

text='面向情绪::房间空调系统有烟味，可考虑做调整'
model.predict(text)
#['设施面向的负面情绪']

輸入輸出改成是繁(正)體中文，輸出三項分類任務資訊

from opencc import OpenCC
t2s = OpenCC('t2s')  # convert from Traditional Chinese to Simplified Chinese
s2t = OpenCC('s2t')  # convert from Simplified Chinese to Traditional Chinese
class_types = ['整体情绪::','面向::','面向情绪::']
def predict(text):
    text = t2s.convert(text)
    response=[]
    for prefix  in class_types:
        response.append(s2t.convert(model.predict(prefix+text)[0]))
    return response

text='位置近市區，人員親切，食物好吃'
predict(text)
#['正面情緒', '服務面向,地點面向', '服務面向的正面情緒,地點面向的正面情緒']