有关NILM中的评价指标的解释请看我们另外一篇博文
,下面简单解释一下代码实现,代码为python版本。
import numpy as np
def tp_tn_fp_fn(states_pred, states_ground): # 定义tp,tn,fp,fn,np.logical_and为逻辑与
tp = np.sum(np.logical_and(states_pred == 1, states_ground == 1),axis = 0).reshape([1,-1]) //
fp = np.sum(np.logical_and(states_pred == 1, states_ground == 0),axis = 0).reshape([1,-1])
fn = np.sum(np.logical_and(states_pred == 0, states_ground == 1),axis = 0