conftest.py 文件源码

def datablock():
    X,y = load_iris(return_X_y=True)
    df = pd.DataFrame(X,columns=['var%d'%i for i in range(4)])
    df['target'] = y

    #make 1 variable categorical
    df['var3'] = df['var3'].apply(lambda x: int(x)).astype(object)

    #make outcome binary
    df['target'].loc[df['target']==2]=1
    return DataBlock(df,df,df,'target')