算法:朴素贝叶斯
import pandas as pd
from collections import defaultdict
import math
train=pd.read_csv("train.txt")
test=pd.read_csv("test.txt")
submit=pd.read_csv("sample_submit.csv")
数据集的样子。
train.head(10)
id | name | gender | |
0 | 1 | 闳家 | 1 |
1 | 2 | 玉璎 | 0 |
2 | 3 | 于邺 | 1 |
3 | 4 | 越英 | 0 |
4 | 5 | 蕴萱 | 0 |
5 | 6 | 子颀 | 0 |
6 | 7 | 靖曦 | 0 |
7 | 8 | 鲁莱 | 1 |
8 | 9 | 永远 | 1 |
9 | 10 | 红孙 | 1
|