需求,我有一列01的label,之前是使用tensorflow直接预测,将输出值使用sigmod之后按照0.5分的,但是效果一般。所以想使用one-hot试试。虽然我也听说二分类不需要one-hot。
操作:直接使用pandas自带的get_dummies()函数。
df_label = pd.get_dummies(df_label)
输出df_label变成若干行两列的dataframe格式。
需求,我有一列01的label,之前是使用tensorflow直接预测,将输出值使用sigmod之后按照0.5分的,但是效果一般。所以想使用one-hot试试。虽然我也听说二分类不需要one-hot。
操作:直接使用pandas自带的get_dummies()函数。
df_label = pd.get_dummies(df_label)
输出df_label变成若干行两列的dataframe格式。