automl.py 文件源码

def baseline(output_dir, basename, valid_num, test_num, target_num):
    preds_valid = np.zeros([valid_num , target_num])
    preds_test = np.zeros([test_num , target_num])

    cycle = 0
    filename_valid = basename + '_valid_' + str(cycle).zfill(3) + '.predict'
    data_io.write(os.path.join(output_dir,filename_valid), preds_valid)
    filename_test = basename + '_test_' + str(cycle).zfill(3) + '.predict'
    data_io.write(os.path.join(output_dir,filename_test), preds_test)