가공에 필요한 함수들

 

기준 값을 0.5로 설정하고, 0.5미만이면 0으로, 0.5 이상이면 1로 바꿔서 표현하기

def cut_off(y,threshold):
    Y = y.copy()
    Y[Y>threshold] = 1
    Y[Y<threshold] = 0
    return(Y.astype(int)) 

pred_Y = cut_off(pred_y, 0.5)
pred_Y