档案分类结果

Question

我有一个有1.000行和两列的csv。第一列是一些文本，第二列是标志(1或0)。我运行了一些二进制分类算法，得到了一些f.score。如何获得输出(保存到文件中)以显示哪些行为1，哪些行为0，以便手动将原始标志与算法的分类进行比较？

Answer 1

假设您有存储在变量中的结果标志(i.e. the prediction that you have made from your binary classification model)，(例如results results)，而保存文本和真标志的数据帧是df，那么您可以执行以下步骤将1-flag texts和0-flag texts放入不同的文件中，如下所示

df['Predicted']=results
df1=df[df['Predicted']==1]
df2=df[df['Predicted']==0]
df1.to_csv('predicted1_flag.csv')
df2.to_csv('predicted0_flag.csv')

如果您提供了代码片段或输出快照，它会变得更好，但是如果您只是想按照类将它们分开，这将很好。