加入列表的大熊猫框架 - 我一直的顺序?

问题描述 投票:2回答:1

所以,我对保险索赔2个脚本的人工神经网络 - 一个脚本是训练/测试和一个执行前进。我与第一个完成,开发利用实际生产数据作为它的测试第二个。目标/类别标签是一个二进制1或0的输入数据是最初的形状(5914,23)的一个数据帧,它是所有数值数据。然后我做在其上的df.values.tolist(),我做的所有值StandardScaler()(比第一个其是权利要求ID其他),并在这个过程中,它通过np.asarray。然后,我通过ANN_Model.Predict_Proba运行它,给了我的5914双概率的列表。现在我想合并回我做了tolist(),所有的概率(称为“predicted_probs”)之前,我有数据帧和这样做的到对原始数据帧新的列(列名为“结果”)和为一类这样做(我只在正班感兴趣)。我通过下面的代码这样做。但我不知道我的结果的顺序是一样的数据框的顺序。是吗?

for i in range (0,len(predicted_probs)):
    original_df["Results"] = pd.Series(predicted_probs[i])
    print (predicted_probs[[i],[1]])

我应该做另一种方式?我必须复制什么是在训练脚本来完成,以期待像对等的结果,因此StandardScaler(),np.asarray等。

提前致谢

python pandas list indices
1个回答
0
投票

你的数据框的形状(5914, 23)ann_model.predict_proba输出5914。因为从你的DF将输出一个连续的单一概率,你可以期望你的结果的顺序是一样的你的数据框的顺序。到正类的概率添加到数据帧,

original_df['Results'] = [i[1] for i in predicted_probs]

还有就是通过predicted_probs不需要你循环

© www.soinside.com 2019 - 2024. All rights reserved.