整数特征的PMI计算

Question

我正在处理一个具有数字特征的数据集，我想计算特征的PMI（逐点互信息）

例如，我有以下几点：

test_features = [[1,1], [1,0], [0,1], [0,0]]
test_labels = [1,1,0,0]

其中目标类为1

在这种情况下，如何计算PMI

编辑：

预期输出：形式为{feature:ppmi}的字典，其中每个特征由其ID表示（在1-hot编码表示中的位置）