【发布时间】:2020-09-12 19:09:52
【问题描述】:
我不知道如何摆脱这个错误。下面是我的示例数据集。我还缺少另一个步骤吗?
Code below:
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
models = RandomForestClassifier(n_estimators=100)
np.random.seed(42)
X = re_arrange.drop('Gender',axis=1)
y = re_arrange['Gender']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
models.fit(X_train,y_train)
models.score(X_test, y_test)
【问题讨论】:
-
RandomForestClassifier要求特征为数字(float或int)。Branch是字符串,不能转换为float。您应该查看分类编码器或单热编码器。 -
@QuangHoang 感谢快速响应。会调查这个
标签: python pandas machine-learning sklearn-pandas