每日一篇學習筆記直到我做完專題 :( [Day43]

huggingface python 新手入門

morris_911 2025-01-11 00:28:33 ‧ 256 瀏覽

分享至

今天終於從期末地獄中解脫又回來繼續學習拉
之前在使用中發現jieba在切割詞的時候不是很精準
並查到可以使用hugging face裡面的一些模型
可以更快速的投入使用所以之後會往那個方向學習
今天先把所有該下載的東西下載好
用一個最簡單的情感分析

from transformers import pipeline

def main():
    # 加載情感分析模型 使用繁體中文
    classifier = pipeline("sentiment-analysis", model="uer/roberta-base-finetuned-jd-binary-chinese")

    # 測試輸入
    input_text = "這次的旅行計劃讓我非常期待！"
    result = classifier(input_text)
    print(result)

if __name__ == '__main__':
    main()