簡介基于樸素貝葉斯算法的新聞分類項目源碼主要面向需要完成Python期末大作業(yè)或課程設計的計算機專業(yè)學生也適合希望在文本分類任務中實戰(zhàn)樸素貝葉斯模型的初學者難度適中能夠覆蓋從數(shù)據預處理到分類評估的完整項目實踐。資源共2000個文件含1996個txt詞頻數(shù)據、3個py腳本及1個md說明文檔壓縮包大小約13.33MBpy腳本分別實現(xiàn)貝葉斯訓練、分類器核心與URL解析等流程txt文件為新聞文本預處理后的特征詞頻表md文檔幫助快速理解項目結構。源碼已經過本地編譯調試可穩(wěn)定運行且作為導師指導下完成的高分大作業(yè)評審得分為98分在分類效果與代碼規(guī)范方面均有較好表現(xiàn)。目前已有154人學習下載讀者可借助源碼與詞頻數(shù)據完整理解樸素貝葉斯從訓練到預測的流程并據此擴展實現(xiàn)其他文本分類場景。1. 樸素貝葉斯做新聞分類為什么這套項目源碼值得你完整跑一遍新聞分類是大多數(shù)人接觸 NLP 的第一個完整項目樸素貝葉斯算法則是這類項目源碼里最常見的基準模型。別看它“樸素”在幾千條標注數(shù)據、類別邊界清晰的中文新聞上它常能和復雜模型打平訓練卻只要幾秒鐘。這套項目源碼做的事很具體用帶類別標簽的新聞文本訓練分類器讓新來的新聞自動歸入財經、體育、娛樂、科技等欄目。源碼里通常包含分詞、特征提取、模型訓練、評估的完整鏈路讀懂它文本分類主流程就走通了一遍。適合剛做完 Python 基礎、想用 sklearn 動手的初學者也適合拿到“樸素貝葉斯新聞分類”作業(yè)、需要改跑源碼的讀者。下文按“原理 → 跑通 → 調參 → 避坑 → 進階”講透。2. 樸素貝葉斯的原理與選型新聞分類為什么首選多項式模型拿到“基于樸素貝葉斯解決新聞分類”的源碼第一件事不是急著跑代碼而是先把模型看懂。很多初學者直接調 sklearn 的 MultinomialNB三行代碼跑完換一個數(shù)據集就翻車根因是沒搞懂這個“樸素”到底樸素在哪、多項式模型又比另外兩個變體好在哪。這一章把原理和選型一次講清。2.1 貝葉斯定理到“樸素”假設新聞分類的數(shù)學底子貝葉斯定理本身不復雜P(類別 | 文本) P(文本 | 類別) × P(類別) / P(文本)我們要的是給定一篇新聞它屬于每個類別的后驗概率取最大者作為預測類別。真正的難點在 P(文本 | 類別)一篇新聞有幾十個詞詞與詞之間的聯(lián)合分布幾乎沒法直接估。樸素貝葉斯在這里做了一步大膽簡化——假設文本里的詞在給定類別條件下互相獨立于是P(文本 | 類別) Π P(詞_i | 類別)拿“湖人隊擊敗勇士隊”舉例簡化后只需要估計 P(“湖人隊” | 體育) × P(“擊敗” | 體育) × P(“勇士隊” | 體育)每個詞的概率都能從訓練語料里數(shù)出來。這個假設在真實語言里顯然不成立“湖人隊”和“勇士隊”出現(xiàn)在同一篇新聞里絕非獨立事件。但有意思的是分類任務只需要比較后驗概率的大小不需要概率值精確所以哪怕假設錯了排序往往仍然正確。這正是樸素貝葉斯“錯了還能打”的底層原因。接下去要選變體。sklearn 提供三個常用實現(xiàn)GaussianNB 面向連續(xù)特征假設特征服從高斯分布BernoulliNB 面向 0/1 布爾特征適合“詞是否出現(xiàn)”的表示MultinomialNB 面向詞頻或 TF-IDF 這類非負數(shù)值假設特征服從多項式分布。新聞文本經過分詞和向量化后得到的是“每個詞在每篇文檔里的重要度”矩陣恰好落在多項式模型的射程內所以項目源碼里幾乎清一色用 MultinomialNB。如果你看到有人用 GaussianNB 處理詞頻矩陣基本可以判斷是誤用——詞頻是離散計數(shù)硬套高斯分布會損失大量分布信息。從概率公式還能看出樸素貝葉斯天生適合新聞分類它把 P(類別) 先驗概率和 P(詞 | 類別) 似然分開建模類別邊界越清晰、各類別用詞差異越大樸素假設造成的誤差越小。財經和科技的高頻詞幾乎不重疊時即使獨立性假設不成立乘積排序依然可靠。這也是為什么在小樣本新聞數(shù)據上樸素貝葉斯經常比詞向量加神經網絡的組合還要穩(wěn)——它不需要海量數(shù)據來擬合詞與詞的交互關系。2.2 中文分詞與特征表示詞頻向量還是 TF-IDF中文新聞沒有天然空格訓練前第一步永遠是分詞。這類源碼最常見的工具是 jieba把整篇新聞切成詞序列過濾停用詞后用空格拼成字符串交給 sklearn 向量化器。分詞粒度直接影響特征質量“湖人隊”被切成“湖人”“隊”還是保持完整取決于 jieba 詞庫和是否加載了自定義詞典。我一般會在項目里維護一個領域詞表把球隊名、人名、產品名用 jieba.add_word() 加進去。特征表示有兩條路線。CountVectorizer 統(tǒng)計詞在每篇文檔的出現(xiàn)次數(shù)直接喂給 MultinomialNB語義上最貼合多項式分布TfidfVectorizer 對詞頻做 IDF 加權壓制“的、了、在”這類高頻但不帶信息的詞還能用 sublinear_tf 對詞頻取對數(shù)壓縮量級。對比項CountVectorizerTfidfVectorizer特征值含義詞在文檔中出現(xiàn)次數(shù)詞頻 × 逆文檔頻率是否適合 MultinomialNB最貼合多項式分布可以特征值非負即可對高頻無意義詞的壓制無靠停用詞表有IDF 自動降權新聞分類實測 F1基準水平通常高 13 個點我的經驗是新聞分類優(yōu)先用 TfidfVectorizer但樸素貝葉斯對兩種表示都不排斥真正影響大的是 max_features。新聞語料詞表動輒幾萬大量低頻詞只出現(xiàn)一兩次對分類沒有貢獻還拖慢訓練。常見做法是把特征維度限制在 500010000配合 min_df2 過濾只出現(xiàn)一次的詞、max_df0.8 過濾出現(xiàn)在八成以上文檔的大路詞。調這三個參數(shù)比糾結 CountVectorizer 還是 TfidfVectorizer 收益大得多。import jieba from sklearn.feature_extraction.text import TfidfVectorizer # 加載停用詞表常見的做法是每行一個詞 with open(stopwords.txt, encodingutf-8) as f: stopwords {line.strip() for line in f} def cut_news(text: str) - str: words [w for w in jieba.cut(text) if w.strip() and w not in stopwords] return .join(words) sample 湖人隊主場擊敗勇士隊詹姆斯砍下三雙 print(cut_news(sample)) # 輸出示例湖人隊 主場 擊敗 勇士隊 詹姆斯 砍下 三雙 vectorizer TfidfVectorizer( max_features8000, min_df2, max_df0.8, sublinear_tfTrue )代碼邏輯說明cut_news 先分詞再過濾空串和停用詞最后空格拼接——不做這一步向量化器會把整句當成一個“詞”特征維度徹底亂掉。TfidfVectorizer 的參數(shù)里max_features 限制詞表規(guī)模min_df 和 max_df 從頻次兩端裁剪sublinear_tf 做詞頻對數(shù)壓縮。這四個參數(shù)是新聞分類特征工程的標配第 4 章還會展開講。3. 把項目源碼跑通從數(shù)據到模型的最小可運行路徑3.1 環(huán)境準備與數(shù)據格式先讓數(shù)據能進模型這類源碼通?；?Python 3.7 以上核心依賴是 scikit-learn、pandas、jieba裝起來沒什么玄學pip 一行解決。真正卡住新手的是數(shù)據格式源碼 README 里可能寫著 THUCNews 數(shù)據集也可能只是一個 data 目錄但模型能吃的只有一種約定——結構化表格。常見做法是把語料整理成 CSV兩列l(wèi)abel 是新聞類別content 是正文UTF-8 編碼。pip install scikit-learn pandas jieba joblib裝完之后先用 pandas 讀一遍數(shù)據打印類別分布。這一步別跳過類別數(shù)量如果差一個數(shù)量級模型會悄悄偏向大類整體準確率看著不錯逐類看 F1 就露餡。我的習慣是把 value_counts() 的結果掛到項目 README 里每次復現(xiàn)先核對數(shù)據再談模型。另一種常見的數(shù)據組織方式是按類別分目錄每個目錄下是一篇篇 .txt 新聞。遇到這種結構需要先寫一個小腳本把目錄結構展平成 CSV再進入下面的流程。我一般會在 utils.py 里放一個 load_data 函數(shù)統(tǒng)一處理 CSV 和目錄兩種輸入返回 DataFrame這樣 train.py 和 predict.py 都不關心數(shù)據原始長什么樣。3.2 訓練主流程分詞、向量化、訓練、評估一條龍下面這段代碼涵蓋了此類源碼的核心鏈路可以直接存成 train.py每步加了注釋方便你對照自己拿到的源碼找差異。import jieba import pandas as pd from sklearn.model_selection import train_test_split from sklearn.feature_extraction.text import TfidfVectorizer from sklearn.naive_bayes import MultinomialNB from sklearn.metrics import classification_report # 1. 讀數(shù)據 df pd.read_csv(news_data.csv, encodingutf-8) df df.dropna(subset[content, label]) print(df[label].value_counts()) # 2. 分詞整篇新聞切成詞序列空格拼回字符串 df[cut] df[content].apply(lambda x: .join(jieba.cut(x))) # 3. 切分先切分再向量化stratify 保持類別比例 X_train, X_test, y_train, y_test train_test_split( df[cut], df[label], test_size0.2, random_state42, stratifydf[label] ) # 4. 向量化fit 只允許出現(xiàn)在訓練集上 vectorizer TfidfVectorizer(max_features8000, sublinear_tfTrue) X_train_vec vectorizer.fit_transform(X_train) X_test_vec vectorizer.transform(X_test) # 5. 訓練alpha 是拉普拉斯平滑系數(shù) model MultinomialNB(alpha1.0) model.fit(X_train_vec, y_train) # 6. 評估 y_pred model.predict(X_test_vec) print(classification_report(y_test, y_pred))邏輯說明分三段。切分階段stratify 讓訓練集和測試集保持與原始數(shù)據一致的類別比例這是對付樣本不平衡的第一道防線很多源碼省略它小類樣本會在切分時被切沒。向量化階段fit_transform 只在訓練集上調用測試集只走 transform兩者順序一旦顛倒就是數(shù)據泄漏第 5 章會專門講。訓練階段alpha 默認 1.0 解決“某個詞在訓練集中沒出現(xiàn)導致概率為零”的零概率問題取值大小直接影響模型保守還是激進。第一次跑通準確率在 0.850.95 都算正常具體看數(shù)據集類別數(shù)和文本質量。如果低于 0.8先別換模型回頭查分詞和停用詞表八成是那兩步出了問題。還有一個自查點把詞表前幾十個詞打出來看如果滿是“的、了、在”這類虛詞說明停用詞沒生效。這段代碼在幾千條新聞、八千維特征上訓練通常在一秒以內慢的是分詞那一步建議把分詞結果緩存到磁盤下次直接讀 cut 列能省下大半時間。3.3 源碼目錄結構一份合格的新聞分類項目長什么樣拿到項目源碼第一步是看目錄不是跑代碼。我見過的最小可用結構是這樣的news_classifier/ ├── data/ # 原始語料CSV 或按類別分好的文本目錄 ├── train.py # 訓練主腳本產物是模型文件 ├── predict.py # 加載模型對單條新聞預測 ├── utils.py # 分詞、停用詞、數(shù)據加載等公共函數(shù) ├── model.pkl # 模型文件注意是否包含向量化器 └── stopwords.txt # 停用詞表每行一個詞判斷源碼是否合格看兩個點。第一predict.py 是否同時加載了向量化器和模型。只保存 model.pkl、丟掉 vectorizer 的源碼不在少數(shù)預測時詞表對不上輸出基本是亂猜。第二train.py 是否固定了隨機種子。不設 random_state每次跑出來的劃分和結果都不一樣你復現(xiàn)不了作者聲稱的準確率還會誤以為是自己環(huán)境的問題。predict.py 的常見寫法長這樣# predict.py 核心邏輯 import joblib import jieba pipe joblib.load(news_pipeline.pkl) def predict_one(text: str) - str: cut_text .join(jieba.cut(text)) return pipe.predict([cut_text])[0] print(predict_one(湖人隊主場擊敗勇士隊詹姆斯砍下三雙)) # 輸出示例體育這段代碼暴露了一個容易忽略的細節(jié)分詞必須和訓練時保持完全一致包括停用詞表、jieba 自定義詞典。如果 predict.py 里重新寫了一套分詞邏輯哪怕只是停用詞加載路徑不同預測效果就會和訓練時對不上。這是“模型文件有了、效果卻復現(xiàn)不了”的最大來源之一。4. 參數(shù)調優(yōu)與效果驗證alpha、特征維度與交叉驗證4.1 alpha 的取值從默認 1.0 到網格搜索alpha 是樸素貝葉斯項目源碼里最值得調的參數(shù)它對應拉普拉斯平滑的強度。訓練時如果某個詞只在個別類別里出現(xiàn)過它在其他類別下的概率會被數(shù)成零乘到整篇新聞的后驗概率上直接把這個類別判死。alpha 給所有詞的計數(shù)統(tǒng)一加一個小數(shù)避免零概率一刀切。alpha 越大各類別下的詞概率越均勻模型越保守alpha 越小模型越依賴訓練集詞頻過擬合風險越高。在新聞分類場景下我的經驗是數(shù)據量大、詞表豐富時alpha 取 0.1 左右往往比默認 1.0 更準數(shù)據量小、文本短時1.0 更穩(wěn)。不要靠感覺猜直接把 alpha 扔進網格搜索from sklearn.model_selection import GridSearchCV param_grid {alpha: [0.01, 0.05, 0.1, 0.5, 1.0, 2.0]} gs GridSearchCV( MultinomialNB(), param_grid, cv5, scoringf1_macro, n_jobs-1 ) gs.fit(X_train_vec, y_train) print(gs.best_params_, gs.best_score_)這里 scoring 我特意選 f1_macro 而不是 accuracy原因很現(xiàn)實新聞數(shù)據里體育類占 60% 的話模型全猜體育也有 60% 準確率但每個類別的 F1 會立刻暴露短板。f1_macro 對每個類別一視同仁地求平均逼著模型把少數(shù)類也學起來。網格搜索跑完后把最優(yōu) alpha 回填到 train.py 重新訓練測試集上再出一版最終報告。注意 GridSearchCV 的 cv5 是在訓練集內部再切五折測試集從頭到尾不能碰否則調參過程同樣會泄漏。4.2 混淆矩陣哪兩類新聞總被搞混分類報告只能告訴你每類的精確率、召回率和 F1但“哪兩類在打架”得看混淆矩陣。新聞分類里最典型的是財經和商業(yè)、科技和數(shù)碼這類邊界模糊的類別高頻詞大量重疊模型分不清很正常?？椿煜仃嚨囊饬x在于——如果錯分集中在一兩對類別你要做的是合并類別或補充語料而不是繼續(xù)調參。import matplotlib.pyplot as plt from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay cm confusion_matrix(y_test, y_pred, labelsmodel.classes_) disp ConfusionMatrixDisplay(confusion_matrixcm, display_labelsmodel.classes_) disp.plot(xticks_rotation45) plt.tight_layout() plt.savefig(confusion_matrix.png, dpi150)讀矩陣有個順序先看對角線再看對角線之外最大的格子。對角線占比高說明整體可靠非對角線上的熱點代表固定的混淆對比如“科技”經常被分到“數(shù)碼”通常是語料來源高度重疊合并成一個類反而讓任務更清晰。保存成圖片放進項目里比每次訓練后打印一屏數(shù)字直觀得多。類別超過十個時記得調大圖片尺寸否則邊緣類別的標簽會擠成一團。我習慣把混淆矩陣和分類報告一起存成 result 目錄后面換模型、換特征都能直接對比。4.3 特征維度與卡方檢驗把詞表選擇交給算法調完 alpha 收益不大時問題大概率在特征工程而不是模型。max_features 從 8000 調到 15000有時候漲一兩個點有時候反而掉取決于語料里有效特征的密度。判斷方法是打印詞表前幾十個詞如果混著大量無關詞說明停用詞沒加夠或 max_df 設得太寬松。比人肉看詞表更可靠的做法是卡方檢驗sklearn 的 chi2 配合 SelectKBest可以直接在向量化之后選出對類別區(qū)分貢獻最大的詞from sklearn.feature_selection import SelectKBest, chi2 selector SelectKBest(chi2, k3000) X_train_sel selector.fit_transform(X_train_vec, y_train) X_test_sel selector.transform(X_test_vec) sel_model MultinomialNB(alpha0.1) sel_model.fit(X_train_sel, y_train) print(sel_model.score(X_test_sel, y_test))卡方檢驗的思路是對每個詞計算它與類別標簽的相關性保留相關性最高的前 k 個。這樣選出來的特征比“憑肉眼刪詞”穩(wěn)定得多尤其適合新聞這種詞表巨大的場景。需要留意的是SelectKBest 同樣只能在訓練集上 fit測試集只做 transform和第 3 章向量化器的規(guī)則一致。如果加了特征選擇后效果反而下降通常是 k 取得太小把有區(qū)分度的詞也裁掉了從 5000 往上加著試。5. 新聞分類源碼落地的五個坑按這個順序排查課程項目的源碼能跑通是一回事移植到自己的數(shù)據上是另一回事。下面五個坑是我在復現(xiàn)這類源碼時真實踩過的每條按“現(xiàn)象 → 原因 → 解決”寫你遇到同類問題可以直接按這個順序排查。5.1 現(xiàn)象小類新聞幾乎全被分錯輸出類別集中在大類原因訓練語料類別嚴重不平衡又沒做分層切分。娛樂類兩萬條、科技類五百條train_test_split 里不寫 stratifydf[label]測試集可能一條科技新聞都分不到模型學到的先驗概率也徹底偏向大類。整體準確率看著挺高逐類看報告就露餡。解決切分時帶上 stratify 參數(shù)。數(shù)據還是不平衡的話對小類做簡單過采樣把文本重復采樣幾份或使用 imbalanced-learn 的 RandomOverSampler。這里有個我當初來回確認過的點sklearn 的 MultinomialNB 沒有 class_weight 參數(shù)想在樸素貝葉斯里直接給類別加權行不通只能從數(shù)據側動手。5.2 現(xiàn)象測試集上效果不錯模型上線后面對新數(shù)據效果崩塌原因源碼把向量化器的 fit_transform 用在了全量數(shù)據上做完特征提取再切訓練測試集。這樣詞表已經包含了測試集的分布信息測試集指標是虛高的新數(shù)據一進來立刻現(xiàn)原形。這是這類源碼里最常見的數(shù)據泄漏寫法不少課程項目都中招。解決嚴格按“先切分、后向量化”的順序執(zhí)行。全流程只允許在訓練集上調用 fit_transform測試集和未來任何新數(shù)據一律只走 transform。檢查代碼只需要看一個位置vectorizer 是在 train_test_split 之前還是之后被 fit 的在之前就是泄漏。注意判斷數(shù)據泄漏最簡單的標準就是看向量化器的 fit 動作發(fā)生在切分之前還是之后。5.3 現(xiàn)象保存模型后predict.py 的預測結果和訓練時對不上原因只保存了模型對象沒保存向量化器。predict.py 里重新 new 了一個 TfidfVectorizer詞表跟訓練時的完全不匹配預測就成了黑匣子亂輸出。這類“模型文件有了、效果卻復現(xiàn)不了”的問題在源碼交接時尤其害人。解決用 joblib.dump 同時保存向量化器和模型更省事的做法是把兩者包成一個 Pipeline 一次保存第 3 章 predict.py 里已經寫好了完整寫法。排查時有個十秒鐘出結果的檢查對比訓練時 vectorizer.get_feature_names_out() 的長度和 predict.py 里新 vectorizer 的長度不一致就說明詞表沒對上。額外提醒一點Pipeline 不會替你分詞predict 入口的文本必須先用和訓練一致的分詞函數(shù)處理否則喂進去的是未切分的原始字符串特征空間直接對不上。5.4 現(xiàn)象中文新聞亂碼或跑出來的特征全是單字和數(shù)字原因數(shù)據文件編碼不是 UTF-8常見于 GBK 編碼的舊語料也可能是 jieba 沒加載自定義詞典人名、球隊名被切成單字特征質量斷崖式下降。這兩個原因看起來毫無關系但癥狀容易被混淆——都表現(xiàn)為模型效果莫名其妙地差。解決先解決編碼pd.read_csv 報編碼錯誤時依次試 utf-8、gbk、gb18030不要一個編碼用到底。再解決分詞把領域詞表通過 jieba.add_word() 加進去然后打印兩篇分詞結果對比確認“湖人隊”保留成整詞而不是“湖人”“隊”。這一步值得最先排查因為它能讓 F1 掉五六個點而且和模型參數(shù)完全無關。排查順序建議先看數(shù)據編碼和分詞輸出再看切分順序最后才是調參數(shù)——順序反了會浪費大量時間。5.5 現(xiàn)象準確率始終卡在 0.75 上不去換模型也沒用原因類別標簽本身有歧義或者語料噪聲大比如娛樂新聞里混著大量明星緋聞和影視資訊邊界本身模糊。這時候不是模型的鍋是任務定義的問題。樸素貝葉斯對標注噪聲比深度學習更敏感因為它是靠詞頻累計的少數(shù)錯標樣本能直接把某個詞的類別傾向帶偏。解決先人工抽 100 條錯分樣本按“標注錯誤”“邊界模糊”“模型真正分錯”三類統(tǒng)計再看要不要清洗數(shù)據。錯分樣本分析是這類項目里最值得做卻最常被跳過的環(huán)節(jié)。如果邊界模糊類占了大頭合并類別或重新定義標簽比換任何模型都有效。6. 進階用增量訓練和置信度閾值把樸素貝葉斯用進生產6.1 增量更新用 partial_fit 跟隨新聞熱點課程項目里的樸素貝葉斯是一次性訓練、一次性評估。真實新聞是持續(xù)產生的關鍵詞和熱點一直在變。每周全量重訓倒不貴麻煩的是向量化器每次重新擬合詞表抖動導致新舊預測結果對不齊。更省力的做法是增量更新MultinomialNB 支持 partial_fit可以直接拿新標注數(shù)據繼續(xù)更新模型但向量化器必須固定住不重訓。# 新到一批已標注新聞按同樣流程分詞后 X_new vectorizer.transform(new_cut_texts) model.partial_fit(X_new, new_labels, classesmodel.classes_)注意第一次調用 partial_fit 必須傳 classes 參數(shù)模型才能確定類別順序。這個做法很適合新聞場景每天用小批量人工修正過的標注樣本喂給模型讓它跟著新詞、新熱點緩慢演化。代價是舊知識衰減慢如果內容風格突變還是需要全量重訓。6.2 置信度閾值讓模型承認自己不確定第二個生產技巧是置信度閾值。新聞分類的錯誤大多集中在“模型自己都不確定”的樣本上與其硬給一個分類不如讓模型承認自己拿不準。用 predict_proba 取出最高概率低于閾值就轉人工審核。proba model.predict_proba(single_vec)[0] if proba.max() 0.6: print(低置信度建議人工審核) else: print(預測類別:, model.classes_[proba.argmax()])閾值取多少我一般從 0.5 起步拿驗證集畫出“閾值 vs 被攔樣本中被人工修正的比例”曲線按業(yè)務能承受的人工量來定。這個技巧不挑模型后面換深度學習模型審核鏈路可以直接復用。我自己的習慣是無論最終要上什么模型第一版一定先用樸素貝葉斯把數(shù)據流、評估口徑、人工審核鏈路打通。它簡單到沒有“調不明白”的借口一旦跑出基準值后面換復雜模型才有對比的錨點。這個習慣幫我在好幾個文本項目里少走了彎路希望幫到你。本文還有配套的精品資源點擊獲取