Pandas 熊猫:如何使用predict_proba获得每个客户的概率
我使用xgboost和objective='binary:logistic'来计算每个客户是否会进行支出的概率。 在sklearn中使用predic_proba将打印0和1的两个概率,如:Pandas 熊猫:如何使用predict_proba获得每个客户的概率,pandas,Pandas,我使用xgboost和objective='binary:logistic'来计算每个客户是否会进行支出的概率。 在sklearn中使用predic_proba将打印0和1的两个概率,如: [[0.56651809 0.43348191] [0.15598162 0.84401838] [0.86852502 0.13147498]] 如何通过pandas插入每个客户ID,以获得如下信息: +----+------------+------------+ | ID | pro
[[0.56651809 0.43348191]
[0.15598162 0.84401838]
[0.86852502 0.13147498]]
如何通过pandas插入每个客户ID,以获得如下信息:
+----+------------+------------+
| ID | prob_0 | prob_1 |
+----+------------+------------+
| 1 | 0.56651809 | 0.43348191 |
| 2 | 0.15598162 | 0.84401838 |
| 3 | 0.86852502 | 0.13147498 |
+----+------------+------------+
可以使用DataFrame()创建表单
list_data = [[0.56651809, 0.43348191],[0.15598162, 0.84401838],[0.86852502, 0.13147498]]
columns = ['prob_0', 'prob_1']
index = [1, 2, 3]
pd.DataFrame(data = list_data, columns = columns, index= index)