Ket5 News Summarizer

🚀 新聞文章摘要模型

本項目基於lcw99/t5-base-korean-text-summary模型微調得到，用於新聞文章的摘要提取，為用戶快速獲取新聞核心內容提供便利。

🚀 快速開始

本模型是基於lcw99創建的lcw99/t5-base-korean-text-summary進行微調得到的“新聞文章摘要模型”。 訓練數據使用了AIHub提供的“文檔摘要文本 (https://aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&aihubDataSe=realm&dataSetSn=97)”中的新聞文章。 後續會持續改進模型性能，使其更加完善。

🔧 技術細節

訓練參數

training_args = Seq2SeqTrainingArguments(
    evaluation_strategy="epoch",                    
    save_strategy="epoch",                          
    save_total_limit=2,                             
    warmup_steps=1000,                             
    learning_rate=5e-5,                             
    per_device_train_batch_size=16,                
    per_device_eval_batch_size=16,               
    num_train_epochs=3,                           
    weight_decay=0.01,                             
    predict_with_generate=True,                    
    fp16=True                               
)

訓練進度

運行環境

• 操作系統：Windows 10
• 顯卡：NVIDIA GeForce RTX 3070, 8192 MiB

框架版本

📄 許可證

本項目採用Apache-2.0許可證。