Multinomial Naive Bayes với làm mịn Laplace. Có K lớp, từ điển kích thước V. Cho ma trận đếm từ countc,v (số lần từ v xuất hiện trong dữ liệu huấn luyện của lớp c), tiên nghiệm P(c), và vector đặc trưng x (số lần mỗi từ xuất hiện trong tài liệu cần phân loại).
Xác suất từ có làm mịn: θc,v=∑v′countc,v′+αVcountc,v+α.
Điểm của lớp c: sc=logP(c)+∑vxvlogθc,v.
Dự đoán lớp có điểm lớn nhất; nếu hòa chọn chỉ số lớp nhỏ hơn. In ra chỉ số lớp.
Dòng 1: K, V, α. Tiếp theo K dòng, mỗi dòng V số đếm của một lớp. Dòng kế: K tiên nghiệm P(c). Dòng cuối: V số (vector x).
1≤K≤50, 1≤V≤100, α>0. Tiên nghiệm dương.
Một số nguyên: chỉ số lớp dự đoán (0-based).
Ví dụ:
Đầu vào:
2 2 1.0
3 1
1 3
0.5 0.5
2 0
Đầu ra:
0
Giải thích:
Đang tải editor...