错误“预期 2D 数组，得到 1D 数组：”

数据挖掘 机器学习 数据集
2021-10-02 04:05:18

在对 Titanic 数据集执行简单的拟合操作时。以下是我的代码：

data = pd.read_csv(r'.\Desktop\DS\Titanic\train.csv')
sex_train = data['Sex'].map({'male':0,'female':1})
survived_train = data['Survived']
sex_survivor_tree = GaussianNB()
sex_survivor_tree.fit(sex_train,survived_train)
1个回答

这有点棘手。使用 pandas 数据，sklearn 只接受类型为 的输入变量（特征）pandas.Dataframe。在您的代码变量类型sex_train中。pandas.Series试试下面的代码：

sex_train = data['Sex'].map({'male':0,'female':1}).to_frame()
其它你可能感兴趣的问题
上一篇nltk中实体识别的标签映射是什么？ 下一篇Softmax 分类器在 LSTM 中永远不允许 100% 的概率？