Esempio n. 1
0
    def __process_evaluated_metric(y1_true, y0_true, y1_hat, y0_hat,
                                   ite_dict, true_ITE_list, predicted_ITE_list, ite_csv_path, iter_id):
        y1_true_np = np.array(y1_true)
        y0_true_np = np.array(y0_true)
        y1_hat_np = np.array(y1_hat)
        y0_hat_np = np.array(y0_hat)

        PEHE = Metrics.PEHE(y1_true_np, y0_true_np, y1_hat_np, y0_hat_np)
        ATE = Metrics.ATE(y1_true_np, y0_true_np, y1_hat_np, y0_hat_np)
        print("PEHE: {0}".format(PEHE))
        print("ATE: {0}".format(ATE))

        true_ATE = sum(true_ITE_list) / len(true_ITE_list)
        predicted_ATE = sum(predicted_ITE_list) / len(predicted_ITE_list)

        Utils.write_to_csv(ite_csv_path.format(iter_id), ite_dict)
        return PEHE, ATE, true_ATE, predicted_ATE