happy transformer

happy transformer

文件與新聞：happytransformer.com

加入我們的 Discord 伺服器：

happy transformer可以輕鬆地使用 NLP Transformer 模型進行微調和執行推理。

3.0.0

1. 用於訓練的 DeepSpeed
2. Apple 用於訓練和推理的 MPS
3. WdB 追蹤訓練跑步狀況
4. 為訓練提供的資料會自動分成幾個部分進行訓練和評估
5. 將模型直接推送到 Hugging Face 的模型中心

在此閱讀完整的 3.0.0 更新，包括重大變更。

任務

任務	推理	訓練
文字生成	✔	✔
文字分類	✔	✔
單字預測	✔	✔
問答	✔	✔
文本到文本	✔	✔
下一句話預測	✔
代幣分類	✔

快速入門

pip install happytransformer

 from happytransformer import HappyWordPrediction
#--------------------------------------#
happy_wp = HappyWordPrediction ()  # default uses distilbert-base-uncased
result = happy_wp . predict_mask ( "I think therefore I [MASK]" )
print ( result )  # [WordPredictionResult(token='am', score=0.10172799974679947)]
print ( result [ 0 ]. token )  # am 

維護者

• Eric Fillion 首席維護者
• 特德·布朗洛 維護者

教學

透過訓練生成文字（GPT-Neo）

文字分類（訓練）

文本分類（仇恨言論檢測）

文本分類（情感分析）

透過訓練進行單字預測（DistilBERT、RoBERTa）

頂級 T5 型號

文法糾正

微調語法校正模型