mtl_dnn_run.py 文件源码

def early_stop_by_valid(valid_auc, k):
    '''Early stop"
    '''

    if k > len(valid_auc):
        return False
    print k, valid_auc

    x = range(k)
    y = valid_auc[-k:]

    # print x, y
    slop, bias = np.polyfit(x, y, 1)
    print "slop: {0:f}".format(slop)

    stopflag = False
    if slop <= FLAGS.slope:
        stopflag = True
    return stopflag