aucCaculate.py

from eda.ml_tools.rocCurve import get_roc_curve
import pandas as pd
from config import *

if __name__ == "__main__":
    test = pd.read_csv(DIRECTORY_PATH + "test{0}.csv".format(TEST_DATE), header=None)
    test_label = test[0].apply(lambda x: x[0]).values
    predict = pd.read_csv(DIRECTORY_PATH +
                          "testset{0}_output_model_{1}-{2}_lr{3}_lambda{4}.txt".format(TEST_DATE,
                            DATA_START_DATE,DATA_END_DATE,"0.03","0.002"), header=None)[0].values
    get_roc_curve(test_label, predict, "1")