classification_classifier_chain.py 文件源码

def binary_trans(trainy, testy, i):
    y1 = trainy['image_path'].apply(lambda x: 1 if x == int(i) else 0).tolist()
    y2 = testy['image_path'].apply(lambda x: 1 if x == int(i) else 0).tolist()
    return(y1, y2)


# <=======================================================================================================>
# load the data