Bernoulli Naive Bayes dùng cho đặc trưng nhị phân. Có K lớp, d đặc trưng. Cho xác suất θc,j=P(xj=1∣c), tiên nghiệm P(c), và vector nhị phân x∈{0,1}d. Điểm của lớp c:
sc=logP(c)+∑j=1d[xjlogθc,j+(1−xj)log(1−θc,j)]
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, d. Tiếp theo K dòng, mỗi dòng d giá trị θc,j. Dòng kế: K tiên nghiệm. Dòng cuối: d giá trị 0/1.
1≤K≤50, 1≤d≤100, 0<θc,j<1, 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
0.9 0.1
0.1 0.9
0.5 0.5
1 0
Đầu ra:
0
Giải thích:
Đang tải editor...