Closed jonahweissman closed 5 years ago
Also, I should note that attack_pred
now just represents the prediction of a datapoint being a member, instead of also including the complement.
i.e. new_attack_pred = old_attack_pred[: , 1]
I see. That makes sense. I will go ahead and merge the pull request.
Because a separate attack model is trained for each output class, the predictions for the attack testing set were grouped by output class, instead of the usual format. I used the indices to place the predictions where they belong.